From 3ca5666a7c9c52f3a36324dcc88864d8d46c026b Mon Sep 17 00:00:00 2001 From: Abhinav Sarje Date: Wed, 26 Jun 2024 21:57:30 +0000 Subject: [PATCH] #0: added transpose shards arg, but whb0 with RM has some flipped grid issue --- .../resnet/tt/ttnn_functional_resnet50_new_conv_api.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/experimental/resnet/tt/ttnn_functional_resnet50_new_conv_api.py b/models/experimental/resnet/tt/ttnn_functional_resnet50_new_conv_api.py index 15b979ae638..9fbfe51a3e7 100644 --- a/models/experimental/resnet/tt/ttnn_functional_resnet50_new_conv_api.py +++ b/models/experimental/resnet/tt/ttnn_functional_resnet50_new_conv_api.py @@ -654,7 +654,7 @@ def run(self, input_tensor, device, batch_size, ops_parallel_config, conv_op_cac # TODO: fix the need to do the reshard here mem_config = ttnn.create_sharded_memory_config_( ttnn.Shape([self.batch_size * x_height * x_width, 64]), - ttnn.CoreGrid(x=7, y=8), + ttnn.CoreGrid(x=8, y=7), ttnn.TensorMemoryLayout.HEIGHT_SHARDED, ttnn.ShardOrientation.ROW_MAJOR, tile_layout=True,