We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 25ca6b8 commit 990d155Copy full SHA for 990d155
1 file changed
test/prototype/moe_training/test_kernels.py
@@ -69,6 +69,7 @@ def _is_sm_10x() -> bool:
69
70
71
@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")
73
def test_row_major_with_jagged_rowwise_scales(round_scales_to_power_of_2: bool):
74
# Tests case where rowwise scales are computed for multiple distinct subtensors,
75
# with end boundary of each group is determine by their end column indexes (offsets).
0 commit comments