Rebased to main (#91)

tenstorrent · Dec 2, 2024 · d770c94 · d770c94
1 parent c3bb21c
commit d770c94
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/TTIR/test_basic_ops.py b/tests/TTIR/test_basic_ops.py
@@ -233,11 +233,12 @@ def module_transpose(a):
     verify_module(module_transpose, input_shapes)
 
 
-def test_scalar_type():
+@pytest.mark.parametrize("input_shapes", [[(3, 3)]])
+def test_scalar_type(input_shapes):
     def module_scalar_type(a):
         return a.shape[0]
 
-    verify_module(module_scalar_type, [(3, 3)])
+    verify_module(module_scalar_type, input_shapes)
 
 
 dim0_cases = []