From ffb30cdc5290701842c2db678cee2bcaf1a46995 Mon Sep 17 00:00:00 2001 From: Michael Melesse Date: Fri, 28 Oct 2022 20:23:11 +0000 Subject: [PATCH] skip ptx assert --- python/test/unit/language/test_core.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/python/test/unit/language/test_core.py b/python/test/unit/language/test_core.py index dfb7f2535..9c1fd7bc8 100644 --- a/python/test/unit/language/test_core.py +++ b/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