From 78bf6ea4439f4c27a17b43d99332fb40c033b481 Mon Sep 17 00:00:00 2001 From: Sankar Manoj Date: Thu, 13 Feb 2025 05:49:26 +0000 Subject: [PATCH] #0: Tests fix --- tests/ttnn/unit_tests/operations/test_new_conv2d.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/ttnn/unit_tests/operations/test_new_conv2d.py b/tests/ttnn/unit_tests/operations/test_new_conv2d.py index ec8bcd3b1e71..05d477aae6f1 100644 --- a/tests/ttnn/unit_tests/operations/test_new_conv2d.py +++ b/tests/ttnn/unit_tests/operations/test_new_conv2d.py @@ -972,7 +972,7 @@ def test_resnet50_conv_wh( pad_w, config_override=config_override, use_shallow_conv_variant=use_shallow_conv_variant, - transpose_shards=use_1d_systolic_array, ## use RM (transpose_mcast=False) with 2D on WH + transpose_shards=True, ## use RM (transpose_mcast=False) with 2D on WH packer_l1_acc=packer_l1_acc, fp32_accum=False, has_bias=has_bias, @@ -1034,7 +1034,7 @@ def test_conv_mem_config_wh( shard_layout=shard_layout, config_override=config_override, use_shallow_conv_variant=use_shallow_conv_variant, - transpose_shards=use_1d_systolic_array, ## use RM (transpose_mcast=False) with 2D on WH + transpose_shards=True, ## use RM (transpose_mcast=False) with 2D on WH packer_l1_acc=True, fp32_accum=False, has_bias=True, @@ -1655,7 +1655,7 @@ def test_unet_conv_wh( config_override, shard_layout=shard_layout, use_shallow_conv_variant=use_shallow_conv_variant, - transpose_shards=use_1d_systolic_array, ## use RM (transpose_mcast=False) with 2D on WH + transpose_shards=True, ## use RM (transpose_mcast=False) with 2D on WH output_layout=output_layout, auto_shard=auto_shard, ) @@ -1754,7 +1754,7 @@ def test_unet_conv_groups_2_wh( config_override, shard_layout=shard_layout, use_shallow_conv_variant=use_shallow_conv_variant, - transpose_shards=use_1d_systolic_array, ## use RM (transpose_mcast=False) with 2D on WH + transpose_shards=True, ## use RM (transpose_mcast=False) with 2D on WH output_layout=output_layout, auto_shard=auto_shard, groups=groups, @@ -1852,7 +1852,7 @@ def test_unet_conv_groups_4_6_wh( config_override, shard_layout=shard_layout, use_shallow_conv_variant=use_shallow_conv_variant, - transpose_shards=use_1d_systolic_array, ## use RM (transpose_mcast=False) with 2D on WH + transpose_shards=True, ## use RM (transpose_mcast=False) with 2D on WH output_layout=output_layout, groups=groups, ) @@ -1951,7 +1951,7 @@ def test_unet_conv_groups_8_wh( config_override, shard_layout=shard_layout, use_shallow_conv_variant=use_shallow_conv_variant, - transpose_shards=use_1d_systolic_array, ## use RM (transpose_mcast=False) with 2D on WH + transpose_shards=True, ## use RM (transpose_mcast=False) with 2D on WH output_layout=output_layout, auto_shard=auto_shard, groups=groups,