mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-06-24 02:14:17 +00:00
addrspace cleanups (#16565)
* addrspace cleanups * bumps * eh, relax a little
This commit is contained in:
parent
2c9d2c0d31
commit
7e6d617935
6 changed files with 12 additions and 24 deletions
|
|
@ -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, 10_000_000)
|
||||
self.assertLessEqual(GlobalCounters.global_ops, 20_000_000)
|
||||
|
||||
def _test_diagonal(self, *shape):
|
||||
a = torch.randn(*shape, dtype=torch.float32, device=device)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue