github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 07:28:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

VALIDATE_WITH_CPU [pr] (#9488)

Browse Source 
* VALIDATE_WITH_CPU [pr]

* fix test
This commit is contained in:
George Hotz
2025-03-18 15:15:04 +08:00
committed by GitHub
parent 935cd01f56
commit 117b7a16ef
8 changed files with 28 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_randomness.py
View File

@@ -99,7 +99,7 @@ class TestRandomness(unittest.TestCase):
@unittest.skipIf(getenv("PTX"), "fails with PTX")
def test_threefry_doesnt_use_long(self):
for ei in lower_schedule(Tensor.rand(20).schedule()):
for (_,ei) in lower_schedule(Tensor.rand(20).schedule()):
if isinstance(ei.prg, CompiledRunner):
for u in ei.prg.p.uops:
self.assertNotIn(u.dtype, {dtypes.long, dtypes.ulong}, msg=f"long found in {ei.prg.p.name}")