#0: Add support for other act types

tenstorrent · Mar 3, 2025 · 4b8e209 · 4b8e209
1 parent b5a81ca
commit 4b8e209
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/ttnn/unit_tests/operations/test_new_conv2d.py b/tests/ttnn/unit_tests/operations/test_new_conv2d.py
@@ -519,8 +519,8 @@ def test_conv_features_multi_device(
 @pytest.mark.parametrize(
     "output_channels, input_channels, input_height, input_width, shard_layout, config",
     (
-        # (256, 256, 8, 8, WS, None),
-        # (128, 128, 32, 32, BS, None),
+        (256, 256, 8, 8, WS, None),
+        (128, 128, 32, 32, BS, None),
         (32, 32, 256, 256, HS, {"act_block_h": 32}),
     ),
 )