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

backend2.py

torch and numpy dtype interop [pr] (#9224)

2025-02-24 18:26:49 +08:00

backend.py

[Bounty] moved index_tensor off cpu in torch_backend (#9916)

2025-04-24 14:12:37 -04:00

example.py

run some tests from torch [pr] (#9252)

2025-02-26 15:42:22 +08: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

test.py

don't modify the ranges on reduce rewrite (#10062)

2025-04-28 12:01:19 -04: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