geohot/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Actions Activity
13,588 Commits 4 Branches 15 Tags
b2e95b2db3675a5f0fcee4c3705103f7ef2afe34
Commit Graph
13588 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
qazalandGitHub b2e95b2db3 rangeify: no copies for write+read of same slice (#16585) 
* failing test

* cleaner failing tests

* assign and read of same slice shouldn't create copies

* err in the changes

* shrink with no overlapping regions in dest is fine
 2026-06-13 02:19:47 +09:00
George HotzandGitHub 833cb37574 move up new style transform (#16592) 
* simpler names

* move up new style transform

* fix that rule
 2026-06-12 10:13:37 -07:00
George HotzandGitHub 51100d2c5c new style cleanups (#16584) 
* spec tighten

* revert

* lin fix

* lin fix

* needed for x86

* revert
 2026-06-12 08:10:38 -07:00
Philip SinitsinandGitHub 76c10cd635 jit: don't memplan buffers reachable from live tensors (#16588) 
The memory planner was suballocating BUFFERs created during JIT capture that are still referenced by external lazy tensor graphs, like the .grad tensors assigned by backward(). The replay then only writes the arena slices, so realizing such a tensor after the call reads freshly allocated memory and silently returns zeros. Hold every BUFFER reachable from a live Tensor instead of only the parameters of the return value; true internals are still planned. Fixes #16571.
 2026-06-12 17:51:54 +03:00
nimlgenandGitHub 2bfdf85f87 hcq2: move pre bufferize (#16589) 
* hcq2: move pre bufferize

* x
 2026-06-12 16:11:59 +03:00
nimlgenandGitHub fb74f75485 var params sort after global params (#16590) 2026-06-12 14:33:15 +03:00
qazalandGitHub 4d34590b7d llama: less E kernels (#16517) 2026-06-12 19:49:25 +09:00
qazalandGitHub 12f4cf0e49 rename amd/test_custom_kernel.py to test_asm_kernel (#16586) 
* rename amd/test_custom_kernel.py to test_asm_kernel

* update
 2026-06-12 16:11:01 +09:00
wozeparrotandGitHub e770805d21 llama: mxfp8 (#16574) 2026-06-11 22:15:24 -07:00
George HotzandGitHub b8aec4cce7 port x86 to new_style (fable slop) and now everything is new style (#16581) 
* port x86 to new_style (fable slop)

* don't change ops

* port NIR to new_style (fable)

* lil cleanup

* fix tests, and remove new_style
 2026-06-11 21:09:34 -07:00
chenyuandGitHub 762f50bd52 move gradient.py to mixin/ [PR] (#16583) 2026-06-11 23:58:21 -04:00
chenyuandGitHub a2cec397f3 UOp cast and bitcast takes DTypeLike [PR] (#16582) 
* UOp cast and bitcast takes DTypeLike [PR]

match Tensor

* fix type
 2026-06-11 22:38:54 -04:00
George HotzandGitHub b97e3e01e3 port NIR to new_style (fable) (#16580) 
* port NIR to new_style (fable)

* lil cleanup
 2026-06-11 18:47:30 -07:00
sirhcmandGitHub 4d893f626a move a bunch of test_schedule to null (#16578) 2026-06-11 20:26:34 -04:00
George HotzandGitHub b57639a6cc port python to new_style (fable) (#16579) 
* port python to new_style (fable)

* doesn't have to be const in python
 2026-06-11 17:26:05 -07:00
George HotzandGitHub a04d2fa4eb port ptx to new_style (fable) (#16577) 
* port ptx to new_style (fable)

* simplify

* simpler
 2026-06-11 17:05:03 -07:00
George HotzandGitHub 587333fddb replace DEFINE_VAR with PARAM (#16576) 
* replace DEFINE_VAR with PARAM

* cleanups

* cleanups
 2026-06-11 15:03:20 -07:00
chenyuandGitHub 5f1e2d3900 PADTO pads Invalids (#16562) 2026-06-11 16:54:26 -04:00
George HotzandGitHub 434a8ffc38 move llvm to new style (#16573) 
* move llvm to new style

* fix wmma

* buffer is early
 2026-06-11 12:59:02 -07:00
George HotzandGitHub 347608a523 put loads back on reg (#16572) 
* put loads back on reg

* fix dsp
 2026-06-11 11:24:50 -07:00
nimlgenandGitHub e5f498de3b hcq2: debug=2 info (#16569) 
* hcq2: debug=2 info

* t

* x

* hcq2: debug=2 info

* x
 2026-06-11 19:52:01 +03:00
qazalandGitHub a83710396c support mselect input to CALL, less kernels in allreduce (#16567) 
* support mselect input to CALL, less kernels in allreduce

* resolve mstack
 2026-06-11 18:10:47 +09:00
qazalandGitHub 7d4a77dce4 relax comma benchmark timeout (#16568) 2026-06-11 18:03:37 +09:00
qazalandGitHub 21f1101691 add allreduce kernel count test (#16566) 2026-06-11 15:54:12 +09:00
wozeparrotandGitHub c38d6a7e3a mxfp8 part 2 (#16561) 2026-06-10 23:36:11 -07:00
sirhcmandGitHub 83971860d8 ci: simplify webgpu install (#16557) 2026-06-10 22:57:19 -04:00
sirhcmandGitHub 6e1b61f16f cleanup some amd deps (#16563) 
don't load hsa runtime, remove ib autogen
 2026-06-10 19:01:56 -04:00
George HotzandGitHub 7e6d617935 addrspace cleanups (#16565) 
* addrspace cleanups

* bumps

* eh, relax a little
 2026-06-10 15:57:18 -07:00
nimlgenandGitHub 2c9d2c0d31 jit: memplan before compile (#16560) 2026-06-10 15:05:15 +03:00
qazalandGitHub 34481830f1 rangeify: fix cost function for AFTER(out, CALL) (#16559) 
* simple failing test

* fix rangeify cost function

* new ops count
 2026-06-10 17:30:50 +09:00
chenyuandGitHub 623b66e0e4 more tensor and mixin cleanups [PR] (#16558) 2026-06-10 00:39:33 -04:00
chenyuandGitHub 7366d32247 getitem cleanups [PR] (#16556) 2026-06-09 22:48:58 -04:00
George HotzandGitHub fd76ac992e cstyle renderer is new style [pr] (#16484) 
* cstyle new style

* switch cstyle renderer to new style

* fix hip

* fixes

* fix webgpu

* correct webgpu is_packed

* fix dsp

* fixes

* fix Ops.RANGE must be CONST

* old style render access

* this is correct

* fix cstyle to good

* dl/dr

* as array

* fix spec

* remove define_local/define_reg

* buffer in shrink

* fix test_tiny

* all tests fix

* param args aren't realized

* wgsl fix

* work

* new gate

* fix opencl qcom

* process replay

* sort order

* fix render index
 2026-06-09 18:36:01 -07:00
sirhcmandGitHub 97d483350c ci: download prebuilt ocelot (#16554) 2026-06-09 19:51:33 -04:00
sirhcmandGitHub f9d88d3c3a fix race in test_quantize_onnx (#16555) 2026-06-09 18:39:48 -04:00
wozeparrotandGitHub 2bdc360606 gemm: mxfp8 hipkittens gemm (#16541) 
* gemm: mxfp8 hipkittens gemm

* feat: update hipkittens

* feat: kernel signature

* clean: just kernel

* feat: from tinygrad

* feat: test

* fix: add back utils

* clean: no diff

* clean: no diff
 2026-06-09 15:20:05 -07:00
chenyuandGitHub 12addee14f tesnor and mixin cleanups [PR] (#16553) 2026-06-09 15:33:13 -04:00
nimlgenandGitHub 2ab2d51099 hcq2: fix repeated calls (#16552) 2026-06-09 19:11:42 +03:00
chenyuandGitHub 3f053a3370 move functional part of rand to RandMixin (#16551) 2026-06-09 09:40:48 -04:00
nimlgenandGitHub fa31c744b9 hcq2: cleaner (#16550) 2026-06-09 16:33:05 +03:00
qazalandGitHub 598cc13ad2 more readable null graph profile in VIZ (#16548) 
* more readable null graph profile in VIZ

* change

* fix flaky test
 2026-06-09 18:35:05 +09:00
qazalandGitHub d18ad49f20 fix flaky test_disktensor (#16549) 2026-06-09 18:23:22 +09:00
qazalandGitHub fa400f9790 less E kernels in all2all (#16546) 2026-06-09 13:51:57 +09:00
qazalandGitHub b8931440ae add all2all schedule test (#16545) 2026-06-09 12:41:35 +09:00
wozeparrotandGitHub 5ef30005fa update hipkittens (#16544) 2026-06-08 18:53:25 -07:00
sirhcmandGitHub 4e2e2e9956 ocelot: use c.DLL (#16540) 2026-06-08 21:27:28 -04:00
chenyuandGitHub 11fee53527 RandMixin [PR] (#16543) 2026-06-08 19:11:28 -04:00
chenyuandGitHub e2ef5cf5c9 no args and kwargs for _multi_like [PR] (#16539) 2026-06-08 17:35:15 -04:00
chenyuandGitHub 12764161c9 UOp.shard support axis=None [PR] (#16538) 
match Tensor
 2026-06-08 11:36:50 -04:00
chenyuandGitHub ebc5390c9a advance indexing to mixin [PR] (#16532) 2026-06-08 09:24:49 -04:00