skip ptx assert

python/test/unit/language/test_core.py

@@ -1329,11 +1329,15 @@ def test_vectorization(N):
         x = tl.load(src + offsets, mask=offsets < N)
         tl.store(dst + offsets, x, mask=offsets < N)
     pgm = _kernel[(1,)](dst, src, N=N, BLOCK_SIZE=src.shape[0])
-    ptx = pgm.asm["ptx"]
-    if N % 16 == 0:
-        assert "ld.global.v4.b32" in ptx
+    if torch.version.hip is None:
+        ptx = pgm.asm["ptx"]
+        if N % 16 == 0:
+            assert "ld.global.v4.b32" in ptx
+        else:
+            assert "ld.global.b32" in ptx
     else:
-        assert "ld.global.b32" in ptx
+        #TODO add rocm assert
+        pass
     # triton.testing.assert_almost_equal(dst, src[:N])
 # ---------------
 # test store