20 commits

Author	SHA1	Message	Date
George Hotz	c331798201	move tests to test/backend (#14691) ... * move tests to test/backend * fix imports * fix CI * revert that one * Fix formatting in README for test command	2026-02-12 11:09:44 +08:00
George Hotz	183d38b128	remove CUSTOM_KERNEL / directly construct it (#14604) ... * remove CUSTOM_KERNEL / directly construct it * clean that up * simpler multi * custom kernel spec * remove Kernel * fix multi * use sharded shape * explicit regression test	2026-02-08 18:43:33 +08:00
George Hotz	03af2404e2	small changes and test fixes from kernel is call (#14586)	2026-02-06 17:08:33 +08:00
George Hotz	6cbcf98627	KernelInfo is required on get_program (#14571) ... * rangeify always adds KernelInfo * fix tests * skip flaky test	2026-02-06 10:49:27 +08:00
chenyu	cfcd1debb5	test schedule with multiple AFTER (#14449)	2026-01-30 15:59:00 -05:00
b1tg	63a1bb8507	multi custom kernel: support input mixed with copy and shard (#13748)	2025-12-29 12:54:27 -05:00
George Hotz	e5a66ace80	multi custom kernel support (#13716) ... * multi custom kernel support * custom kernel xfrom * works * no SPEC=2 on ck * panic * touchups	2025-12-16 11:36:30 -04:00
ayanhan	92b40290c7	fix: add test_sum_int and remove outdated TODO in test_custom_kernel (#13560)	2025-12-03 21:51:58 -05:00
George Hotz	295600dc5a	saturday coffee shop work parsing the att format (#13295) ... * saturday coffee shop work parsing the att format * add examples * parser * classes of packets * fully vibe coded parser * vibing * empty * some vibe names * vibes * most of these are wrong * more vibes * better names * parsing * parse * cleanup parser * touchups	2025-11-16 08:25:51 -08:00
chenyu	bfb0c0391f	test custom eye function (#13134) ... this version is also faster with NOOPT	2025-11-06 14:51:55 -05:00
chenyu	f33c182393	test custom qkv kernel (#13118) ... adding the online softmax hits infinite loop so starting with this	2025-11-05 23:32:13 -05:00
George Hotz	1e3d6e49a6	index slicing + allclose (#13071) ... * continue work on slicing+allclose * Revert "Revert "slicing + allclose"" This reverts commit 6c7a12f21c. * fix tests + better syntax * forgot an after * slot is an integer	2025-11-03 13:01:48 +08:00
George Hotz	6c7a12f21c	Revert "slicing + allclose" ... This reverts commit c9a1e35b1e.	2025-11-03 12:05:44 +08:00
George Hotz	c9a1e35b1e	slicing + allclose	2025-11-03 12:00:45 +08:00
chenyu	bebec73471	write custom_sum with set and after (#13045)	2025-11-01 10:45:30 -04:00
George Hotz	e98506735b	add CONTRACT support to UOp programs (#13043) ... * add contract support * use contract * 342 tflops	2025-11-01 19:11:32 +08:00
chenyu	f396df26ea	test custom sum (#13039) ... * test custom sum this is higher level than set and after? * only float	2025-10-31 19:25:56 -04:00
George Hotz	e066b3176b	hotfix: types and names for custom kernel test	2025-10-31 17:34:55 +08:00
George Hotz	54f48f93c6	working backward pass in custom kernel (#13032) ... * working backward pass in custom kernel * custom_kernel tensor method * no SPEC=2	2025-10-31 17:26:18 +08:00
George Hotz	b791d70725	support custom UOp kernels (#13028) ... * support custom UOp kernels * no number * multioutput works * backward kernel runs * move kernel class * grad later * work * no tags in kernel graph * test arange * arange + contig * delete comment	2025-10-31 15:51:39 +08:00