Skip to content

Commit 0219747

Browse files
committed
Revert "run test trans reshape but ensure that no convert layouts exist on xpu"
This reverts commit ccb3953. Does not work on LTS currently, as we are removing tensor pointers on LTS and producing different TTGIR.
1 parent ccb3953 commit 0219747

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

python/test/unit/language/test_core.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -4262,6 +4262,11 @@ 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+
42654270
@triton.jit
42664271
def kernel(in_base_ptr, out_base_ptr, IN_SHAPE0: tl.constexpr, IN_SHAPE1: tl.constexpr):
42674272

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

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

42964297
np.testing.assert_equal(to_numpy(expected), to_numpy(actual))
42974298

0 commit comments

Comments
 (0)