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
Roelof van Dijk 0dc2ff431d
 fix: revive torch backend (#13280) 
* fix: revive torch backend

* as_strided view vs copy

* Revert "as_strided view vs copy"

This reverts commit 82a61223f2.

* add extra tests (move inplace, add fusion tests)

* better fusion with inplace_op

* no optimizer hooks (break mnist training fusion)

* split off fusion tests in separate file, assert on resnet fusion

fix: remove comments

* cleanup, reduce diff

* reduce diff

* better fusion and identity checks

---------

Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
2025-11-19 15:26:50 -08:00
..
.gitignore torch backend works for ResNet-18 (#9200) 2025-02-22 22:16:23 +08:00
backend.py fix: revive torch backend (#13280) 2025-11-19 15:26:50 -08:00
backend2.py torch and numpy dtype interop [pr] (#9224) 2025-02-24 18:26:49 +08:00
example.py fix: revive torch backend (#13280) 2025-11-19 15:26:50 -08:00
test.py fix: revive torch backend (#13280) 2025-11-19 15:26:50 -08:00
test_compile.py start torch.compile support (#9279) 2025-02-27 10:29:51 +08:00
test_inplace.py move frontend dir to nn [pr] (#12470) 2025-10-07 10:42:22 +08:00
test_kernel_fusion.py fix: revive torch backend (#13280) 2025-11-19 15:26:50 -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 fix: revive torch backend (#13280) 2025-11-19 15:26:50 -08:00