Fix for test for reduce ops

intel · Jun 25, 2024 · 3455be9 · 3455be9
1 parent 3c27988
commit 3455be9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/python/test_op.py b/test/python/test_op.py
@@ -195,7 +195,7 @@ def test_concatenation(batch, hidden_dim, tensors, axis):
 
 
 @pytest.mark.parametrize("batch", [16, 128])
-@pytest.mark.parametrize("hidden_dim", [256, 512])
+@pytest.mark.parametrize("hidden_dim", [128, 256])
 @pytest.mark.parametrize("axis", [0, 1, -1, -2, None])
 @pytest.mark.parametrize(
     "op", [torch.max, torch.mean, torch.min, torch.prod, torch.sum]