tinygrad/extra/torch_backend
geohotstan 309afa20b7
add Tensor.max_unpool2d (#9518)
* why does max_unpool2d feel slower than out.gradient ...

* slightly cleaner

* what happened to ruff

* need to think about this some more

* slightly faster now?

* clean up, 1 more failing edge case

* ok good

* working TINY_BACKEND

* nit doc wording

* retry CI
2025-03-22 12:11:33 -04:00
..
.gitignore torch backend works for ResNet-18 (#9200) 2025-02-22 22:16:23 +08:00
backend.py add Tensor.max_unpool2d (#9518) 2025-03-22 12:11: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 add masked_select to tensor.py (#9468) 2025-03-17 16:05:36 -04:00
test_compile.py start torch.compile support (#9279) 2025-02-27 10:29:51 +08:00
test_inplace.py torch: support in-place operations on views (#9371) 2025-03-10 23:29:00 +08:00
test_multigpu.py torch backend multigpu - add devices and tests (#9414) 2025-03-12 11:33:11 +08:00
torch_tests.py rig up torch's testing framework [pr] (#9254) 2025-02-26 18:46:22 +08:00
wrapped_tensor.cpp torch backend multigpu - add devices and tests (#9414) 2025-03-12 11:33:11 +08:00