tinygrad/extra/torch_backend
Daniel d65bd669f8
update tiny torch backend hook (#12575)
* update the backend to fix torch deprecation warning

* use param_hook to avoid full backward hook needlessly firing on inputs which do not require gradients

* fix indentation

---------

Co-authored-by: chenyu <chenyu@fastmail.com>
2025-10-15 14:02:33 -04:00
..
.gitignore torch backend works for ResNet-18 (#9200) 2025-02-22 22:16:23 +08:00
backend.py update tiny torch backend hook (#12575) 2025-10-15 14:02:33 -04:00
backend2.py torch and numpy dtype interop [pr] (#9224) 2025-02-24 18:26:49 +08:00
example.py run some tests from torch [pr] (#9252) 2025-02-26 15:42:22 +08:00
test.py remove FUSE_ARANGE (#12511) 2025-10-08 04:54:07 -04: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_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 rename lazydata to uop (#10698) 2025-06-08 08:42:22 -07:00