Reduce dimensionality of inputs, to work under default max-dim=4

nv-legate · May 31, 2024 · 879bc31 · 879bc31
1 parent e7105f8
commit 879bc31
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/tests/integration/test_matmul.py b/tests/integration/test_matmul.py
@@ -47,7 +47,6 @@ def operation(lib, *args, **kwargs):
 @pytest.mark.parametrize(
     "a_shape",
     (
-        (2, 3, 4, 5),
         (3, 4, 5),
         (4, 5),
         (5,),
@@ -56,7 +55,6 @@ def operation(lib, *args, **kwargs):
 @pytest.mark.parametrize(
     "b_shape",
     (
-        (2, 3, 5, 6),
         (3, 5, 6),
         (5, 6),
         (5,),
@@ -73,7 +71,6 @@ def test_operator(a_shape, b_shape):
 @pytest.mark.parametrize(
     "a_shape",
     (
-        (2, 3, 4, 5),
         (3, 4, 5),
         (4, 5),
         (5,),
@@ -82,7 +79,6 @@ def test_operator(a_shape, b_shape):
 @pytest.mark.parametrize(
     "b_shape",
     (
-        (2, 3, 5, 5),
         (3, 5, 5),
         (5, 5),
     ),