Skip to content

Commit 99b82c6

Browse files
committed
only skip the asm check
1 parent 0219747 commit 99b82c6

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

python/test/unit/language/test_core.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4262,11 +4262,6 @@ def kernel():
42624262

42634263
def test_trans_reshape(device):
42644264

4265-
if is_xpu():
4266-
pytest.skip(
4267-
"test_trans_reshape: XPU block pointer implementation preserves the block pointer load for eventual lowering to 2D block read, does not convert layouts in TTGIR"
4268-
)
4269-
42704265
@triton.jit
42714266
def kernel(in_base_ptr, out_base_ptr, IN_SHAPE0: tl.constexpr, IN_SHAPE1: tl.constexpr):
42724267

@@ -4291,8 +4286,9 @@ def kernel(in_base_ptr, out_base_ptr, IN_SHAPE0: tl.constexpr, IN_SHAPE1: tl.con
42914286
actual = torch.zeros(expected.shape, dtype=torch.int32, device=device)
42924287

42934288
k = kernel[(1, )](input, actual, shape[0], shape[1])
4294-
assert k.asm['ttgir'].count(
4295-
'triton_gpu.convert_layout') == 1, "Expected exactly one convert_layout op in the TTGIR after optimization"
4289+
if not is_xpu():
4290+
assert k.asm['ttgir'].count(
4291+
'triton_gpu.convert_layout') == 1, "Expected exactly one convert_layout op in the TTGIR after optimization"
42964292

42974293
np.testing.assert_equal(to_numpy(expected), to_numpy(actual))
42984294

0 commit comments

Comments
 (0)