#8131: temp fix for PCC issue on W0.

tenstorrent · May 30, 2024 · e75540b · e75540b
1 parent ae26f1e
commit e75540b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/models/experimental/resnet/tt/ttnn_functional_resnet50.py b/models/experimental/resnet/tt/ttnn_functional_resnet50.py
@@ -640,6 +640,7 @@ def __call__(self, input_tensor) -> ttnn.Tensor:
         x = ttnn.reshape(x, (1, 1, 56 * 56 * self.batch_size, 64))
         if is_wormhole_b0():
             # TODO: fix the need to do the reshard here
+            x = ttnn.to_memory_config(x, ttnn.L1_MEMORY_CONFIG)  # 8952 Remove this after fixing the issue
             x = ttnn.to_memory_config(x, self.layer1_module1.conv1.conv.input_sharded_memory_config)
 
         x = ttnn.to_layout(x, ttnn.TILE_LAYOUT, dtype=self.model_config["ACTIVATIONS_DTYPE"])