Skip to content

Commit 990d155

Browse files
authored
[ROCm] test(moe): skip jagged rowwise scales test. (#4304)
1 parent 25ca6b8 commit 990d155

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

test/prototype/moe_training/test_kernels.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,7 @@ def _is_sm_10x() -> bool:
6969

7070

7171
@pytest.mark.parametrize("round_scales_to_power_of_2", [True, False])
72+
@skip_if_rocm("jagged rowwise scales kernel vs torch reference mismatch on ROCm")
7273
def test_row_major_with_jagged_rowwise_scales(round_scales_to_power_of_2: bool):
7374
# Tests case where rowwise scales are computed for multiple distinct subtensors,
7475
# with end boundary of each group is determine by their end column indexes (offsets).

0 commit comments

Comments
 (0)