feat: support 1D vectors in MatMul

closes #948
2026-02-09 20:25:34 -05:00 · 2021-12-13 16:36:08 +01:00
parent 214eebb458
commit 94b9d83cbb
4 changed files with 47 additions and 4 deletions
--- a/tests/numpy/test_compile.py
+++ b/tests/numpy/test_compile.py
@@ -1338,6 +1338,21 @@ def test_compile_and_run_constant_dot_correctness(
            (1, 2),
            (0, 8),
        ),
+        pytest.param(
+            (2,),
+            (2,),
+            (0, 8),
+        ),
+        pytest.param(
+            (5, 5),
+            (5,),
+            (0, 4),
+        ),
+        pytest.param(
+            (5,),
+            (5, 5),
+            (0, 4),
+        ),
    ],
 )
 def test_compile_and_run_matmul_correctness(