#12662: adjust shapes to be tile sized for max unit test

tenstorrent · Dec 20, 2024 · 51ecc98 · 51ecc98
1 parent a4b4d79
commit 51ecc98
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/ttnn/unit_tests/operations/test_max.py b/tests/ttnn/unit_tests/operations/test_max.py
@@ -95,9 +95,9 @@ def test_max_global(device, batch_size, h, w):
 @pytest.mark.parametrize(
     "input_shape_and_dim",
     [
-        ((2, 32, 64, 64), -4),
-        ((2, 32, 64, 64), -3),
-        ((2, 32, 64), -3),
+        ((32, 32, 32, 64), -4),
+        ((2, 32, 32, 64), -3),
+        ((32, 32, 64), -3),
     ],
 )
 @pytest.mark.parametrize("keepdim", [True, False])