Skip to content

Commit 4e0281e

Browse files
authored
[AMD] Enable passing test_reduce_layouts (#7126)
1 parent baa03c6 commit 4e0281e

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

python/test/unit/language/test_core.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3078,12 +3078,8 @@ def test_reduce_layouts(M, N, src_layout, axis, epilogue_kind, dtype_str, add_ov
30783078
if isinstance(src_layout,
30793079
(MfmaLayout, MmaLayout)) and (M < src_layout.instr_shape[0] or N < src_layout.instr_shape[1]):
30803080
pytest.skip("Skipping because tensor shape is smaller than M(f)maLayout instr_shape")
3081-
if is_hip() and isinstance(src_layout, MfmaLayout) and ((M, N) == (128, 128)):
3082-
pytest.skip("Skipping test because it runs out of shared memory")
30833081
if reduce_op == "sum" and dtype_str == "float16" and M * N > 1024:
30843082
pytest.skip("Skipping sum reduction on float16 due to accuracy issues")
3085-
if is_hip() and isinstance(src_layout, LinearLayout):
3086-
pytest.skip("FIXME: LinearLayout not supported on HIP")
30873083

30883084
if isinstance(src_layout, MmaLayout) and src_layout.version == 3:
30893085
src_layout.instr_shape[2] = 16 if dtype_str == "float16" else 8

0 commit comments

Comments
 (0)