Skip test for aten::mm when input is sized 0 (#1421)

values of matmul of [m, 0] and [0, n] matrices are undefined.
microsoft · Apr 23, 2024 · c979aae · c979aae
1 parent 75d36cb
commit c979aae
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/function_libs/torch_lib/ops_test_data.py b/tests/function_libs/torch_lib/ops_test_data.py
@@ -1024,7 +1024,10 @@ def _where_input_wrangler(
         reason="fixme (core dump): ORT aborts on scalar inputs to Reduce*-18. https://github.com/microsoft/onnxruntime/issues/16492",
     ),
     TorchLibOpInfo("minimum_bool", core_ops.aten_minimum_bool),
-    TorchLibOpInfo("mm", core_ops.aten_mm),
+    TorchLibOpInfo("mm", core_ops.aten_mm).skip(
+        matcher=lambda sample: torch.numel(sample.input) == 0,
+        reason="values of matmul of [m, 0] and [0, n] matrices are undefined",
+    ),
     TorchLibOpInfo("mT", core_ops.aten_mT),
     TorchLibOpInfo("mT", core_ops.aten_mT_complex, complex=True),
     TorchLibOpInfo("mul", core_ops.aten_mul),