mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-10 23:48:01 -05:00
fix test_tensor_cores_codegen for ptx renderer (#10119)
This commit is contained in:
@@ -1084,6 +1084,8 @@ class TestLinearizer(unittest.TestCase):
|
||||
assert "0x201000" in prg.src
|
||||
elif Device.DEFAULT == "AMD" and getenv("AMD_LLVM", 0):
|
||||
assert "@llvm.amdgcn.wmma" in prg.src
|
||||
elif Device[Device.DEFAULT].renderer.suffix == "PTX":
|
||||
assert "mma.sync.aligned" in prg.src
|
||||
else:
|
||||
assert "__WMMA_" in prg.src
|
||||
|
||||
|
||||
Reference in New Issue
Block a user