#8967: fix padded shape

tenstorrent · May 30, 2024 · 1ca6348 · 1ca6348
1 parent fc9a458
commit 1ca6348
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ttnn/ttnn/operations/core.py b/ttnn/ttnn/operations/core.py
@@ -284,7 +284,7 @@ def from_torch(
             memory_config = ttnn.DRAM_MEMORY_CONFIG
         tensor = ttnn.to_device(tensor, device, memory_config=memory_config)
 
-    if shape_with_padding is not None and shape_with_padding != tensor.shape and mesh_mapper is None:
+    if shape_with_padding is not None and shape_with_padding != tensor.shape:
         tensor = ttnn.reshape(tensor, shape_with_padding)
 
     return tensor