Update test_disasm.py

triton-lang · Sep 20, 2024 · b36cf6c · b36cf6c
1 parent 42f4adb
commit b36cf6c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/python/test/unit/tools/test_disasm.py b/python/test/unit/tools/test_disasm.py
@@ -3,7 +3,6 @@
 import triton
 import pytest
 import triton.language as tl
-from triton.tools.disasm import get_sass
 
 
 def test_disam_cubin():
@@ -17,6 +16,6 @@ def kernel(X, i: tl.constexpr):
     x = torch.empty(1, dtype=torch.int32, device='cuda')
     h = kernel[(1, )](x, i=12)
     assert x[0] == 12
-    sass = get_sass(h.asm["cubin"])
+    sass = h.asm["sass"]
     # check that the sass has a store instruction.
     assert "STG.E" in sass