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

addrspace special/range (#16647)

Browse source 
* addrspace special/range

* just include indexing

* define var is alu

* bring old ignore indexing back

* mults to fix

* fixes

* ALU

* fixes
This commit is contained in:
George Hotz 2026-06-17 15:57:37 -07:00 • committed by GitHub
commit aef85ddc4d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 26 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

2
extra/torch_backend/test.py
View file

@ -248,7 +248,7 @@ class TestTorchBackend(unittest.TestCase):
samples = torch.randint(0, X_train.shape[0], (32,))
X,Y = X_train[samples], Y_train[samples]
X.cpu(), Y.cpu()
self.assertLessEqual(GlobalCounters.global_ops, 20_000_000)
self.assertLessEqual(GlobalCounters.global_ops, 25_000_000)
def _test_diagonal(self, *shape):
a = torch.randn(*shape, dtype=torch.float32, device=device)