tinygrad/extra/torch_backend
George Hotz 690dac79b5
don't modify the ranges on reduce rewrite (#10062)
* bug in div range folding

* simpler

* oh, this is right for indexing, but the div mod folding needs to be fixed

* reenable

* Passing test_complexity_w_unroll2 (#10068)

* Passing

* remove non_folded_divs

* Add check for negative tern in div folding

* Add test

* bump that limit

* fix casted

---------

Co-authored-by: Sieds Lykles <93992551+S-Lykles@users.noreply.github.com>
2025-04-28 12:01:19 -04:00
..
.gitignore torch backend works for ResNet-18 (#9200) 2025-02-22 22:16:23 +08:00
backend.py [Bounty] moved index_tensor off cpu in torch_backend (#9916) 2025-04-24 14:12:37 -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 don't modify the ranges on reduce rewrite (#10062) 2025-04-28 12:01:19 -04:00
test_compile.py start torch.compile support (#9279) 2025-02-27 10:29:51 +08:00
test_inplace.py better fix inplace detach (#9557) 2025-03-24 22:50:28 +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: partial fix for strided related test fails (#9642) 2025-03-31 05:45:18 -04:00