11 Commits

Author	SHA1	Message	Date
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