mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity
tinygrad/extra/torch_backend
History
Exact
George Hotz d1cce7a476
 put the ranges on store instead of after (#15759) 
* put the ranges on store instead of after

* better assert

* fix stuff

* comment out slow rules i don't understand

* simpler rule

* closer

* return false for store

* fix loop

* only a few schedule failures remain

* remove stores to self

* all tests pass locally

* remove junk

* regression test and fix

* better test, bump broken torch count

* bugfix with regression test

* new fusion is better
2026-04-16 19:06:40 +08:00
..
.gitignore torch backend works for ResNet-18 (#9200) 2025-02-22 22:16:23 +08:00
backend.py support shift by Tensor (#15623) 2026-04-06 15:14:57 -04:00
backend2.py torch and numpy dtype interop [pr] (#9224) 2025-02-24 18:26:49 +08:00
example.py put the ranges on store instead of after (#15759) 2026-04-16 19:06:40 +08:00
test.py fix TINY_BACKEND=1 cumsum (#14138) 2026-01-14 09:54:49 -05:00
test_compile.py start torch.compile support (#9279) 2025-02-27 10:29:51 +08:00
test_inplace.py remove allow_shape_mismatch in Tensor.replace (#14536) 2026-02-04 12:38:18 -05:00
test_kernel_fusion.py put the ranges on store instead of after (#15759) 2026-04-16 19:06:40 +08:00
test_multigpu.py move frontend dir to nn [pr] (#12470) 2025-10-07 10:42:22 +08:00
torch_tests.py rig up torch's testing framework [pr] (#9254) 2025-02-26 18:46:22 +08:00
wrapped_tensor.cpp remove allow_shape_mismatch in Tensor.replace (#14536) 2026-02-04 12:38:18 -05:00