github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 06:58:11 -05:00
Code Issues Packages Projects Releases Wiki Activity

hotfix: linearizer test fixup

Browse Source
This commit is contained in:
George Hotz
2024-06-28 10:52:25 -07:00
parent c9714dfcf4
commit 80ac21200b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_linearizer.py
View File

@@ -441,7 +441,7 @@ class TestLinearizer(unittest.TestCase):
assert accs[0].dtype == stores[0].src[-1].dtype == dtypes.float.vec(4)
assert stores[0].src[0].op is UOps.DEFINE_LOCAL
# the second store is to gds with no upcasts
assert accs[1].dtype == stores[1].src[-1].dtype == dtypes.float
assert accs[1].dtype == stores[1].src[2].dtype == dtypes.float
assert stores[1].src[0].op is UOps.DEFINE_GLOBAL
@unittest.skipIf(CI and Device.DEFAULT in {"AMD"}, "AMD CI doesn't support multiple sync threads yet")
@@ -794,7 +794,7 @@ class TestLinearizer(unittest.TestCase):
assert stores[0].src[-1].op is not UOps.CAST
# the global store doesn't change
assert stores[1].src[-1].dtype == dtypes.float
assert stores[1].src[2].dtype == dtypes.float
@unittest.skipUnless(Device[Device.DEFAULT].renderer.supports_float4, "test requires float4")
def test_skip_unmatching_upcasts(self):