github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity
11,618 Commits 203 Branches 13 Tags
7a81a3cb98dfdb2801e1c4bafde35dc45d239d2d
Commit Graph

15 Commits

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