#0: added transpose shards arg, but whb0 with RM has some flipped gri…

…d issue
tenstorrent · Jun 27, 2024 · 3ca5666 · 3ca5666
1 parent 41d7e34
commit 3ca5666
Showing 1 changed file with 1 addition and 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
@@ -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,