[Core] Debug unaligned MXFP8 dequantize tests (#1450)

* Skip MXFP8 dequantize tests with invalid alignment Signed-off-by: Tim Moon <tmoon@nvidia.com> * Remove test case with unaligned rows Signed-off-by: Tim Moon <tmoon@nvidia.com> --------- Signed-off-by: Tim Moon <tmoon@nvidia.com>
NVIDIA · Feb 1, 2025 · f694199 · f694199
1 parent f5f2872
commit f694199
Showing 1 changed file with 8 additions and 0 deletions.
diff --git 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) {