fix test_tensor_cores_codegen for ptx renderer (#10119)

This commit is contained in:
Ignacio Sica
2025-05-01 21:52:36 -03:00
committed by GitHub
parent 30bd6a619f
commit 8f79492c75

View File

@@ -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