github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

remove the all_int(shape) check in Tensor._loadop (#2874)

Browse Source 
* remove the all_int(shape) check in Tensor._loadop

we can support jittable symbolic shape random with custom rand now, and we can formalize it in the test after threefry is ready

* MOCKHIP false positive
This commit is contained in:
chenyu
2023-12-20 15:04:50 -05:00
committed by GitHub
parent 5f07ef455e
commit e13b4964d7
2 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
test/test_symbolic_ops.py
View File

@@ -46,6 +46,7 @@ class TestSymbolicOps(unittest.TestCase):
expected = f(q, k, v).numpy()
np.testing.assert_allclose(symbolic, expected, atol=1e-6, rtol=1e-6)
@unittest.skipIf(getenv("MOCKHIP"), "MOCKHIP only compiles and does not run")
def test_attention_training(self):
with Tensor.train():
self.test_attention(dropout_p=0.0)