nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity
13,597 Commits 334 Branches 15 Tags
bcdb988df0231cd9dbcac55fa3a5630d2d10784a
Commit Graph
13597 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
sirhcmandGitHub bcdb988df0 split comma benchmark, dsp on c4 [PR] (#16598) 2026-06-12 23:26:05 -04:00
George HotzandGitHub 6b8fdfe4ca alu addrspace is where the math happens (#16606) 
* alu addrspace

* fix cstyle/llvm

* on ptx, reg+alu are the same thing
 2026-06-12 20:01:28 -07:00
wozeparrotandGitHub 67a4f129c2 llama: fix bf16 gemm oob (#16603) 2026-06-12 19:43:05 -07:00
sirhcmandGitHub 8862c7549c new-style dcache_flush (#16602) 2026-06-12 22:25:08 -04:00
chenyuandGitHub 9e72a6b376 more indexing cleanup [PR] (#16600) 2026-06-12 21:33:47 -04:00
chenyuandGitHub aa32d309db fix rangeify indexing for pad/reduce (#16599) 2026-06-12 20:26:15 -04:00
George HotzandGitHub 96b86aad7b move new style transform up more (#16593) 
* move new style transform up more

* pm_move_gates_from_index works on new style
 2026-06-12 17:20:12 -07:00
chenyuandGitHub a35964493e UPat method cleanups [PR] (#16596) 2026-06-12 17:22:54 -04:00
chenyuandGitHub 3036b15ed9 remove Tensor.ufix [PR] (#16594) 
* remove Tensor.ufix [PR]

* inline _ufix_keep_dtype
 2026-06-12 14:40:28 -04:00
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