From f6941996f02167a0bf61ded1d33c53dfe13fb6b1 Mon Sep 17 00:00:00 2001 From: Tim Moon <4406448+timmoon10@users.noreply.github.com> Date: Sat, 1 Feb 2025 14:26:52 -0800 Subject: [PATCH] [Core] Debug unaligned MXFP8 dequantize tests (#1450) * Skip MXFP8 dequantize tests with invalid alignment Signed-off-by: Tim Moon * Remove test case with unaligned rows Signed-off-by: Tim Moon --------- Signed-off-by: Tim Moon --- tests/cpp/operator/test_dequantize_mxfp8.cu | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tests/cpp/operator/test_dequantize_mxfp8.cu b/tests/cpp/operator/test_dequantize_mxfp8.cu index c24a739b81..1a090c3a5c 100644 --- a/tests/cpp/operator/test_dequantize_mxfp8.cu +++ b/tests/cpp/operator/test_dequantize_mxfp8.cu @@ -402,6 +402,14 @@ TEST_P(DequantizeMXFP8TestSuite, TestDequantizeMXFP8) GTEST_SKIP(); } + // Skip cases with invalid alignment + if (rowwise && tensor_size.second % 32 != 0) { + GTEST_SKIP(); + } + if (colwise && tensor_size.first % 32 != 0) { + GTEST_SKIP(); + } + TRANSFORMER_ENGINE_TYPE_SWITCH_FP8_ONLY(input_type, InputType, TRANSFORMER_ENGINE_TYPE_SWITCH_FP16_FP32_ONLY(output_type, OutputType, if (quantize_then_dequantize) {