rearrange parameters

tenstorrent · Jan 16, 2025 · bf1a98c · bf1a98c
1 parent 9b54011
commit bf1a98c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/runtime/lib/ttnn/operations/ccl/mesh_shard.cpp b/runtime/lib/ttnn/operations/ccl/mesh_shard.cpp
@@ -49,9 +49,10 @@ void FullToShardShape(const ::ttnn::Tensor &input, ::ttnn::Tensor &out,
     }
 
     out = ::ttnn::distributed::distribute_tensor(
-        input, meshDevice,
+        input,
         *::ttnn::distributed::shard_tensor_to_2d_mesh_mapper(
-            meshDevice, meshDevice.shape(), shard2dConfig));
+            meshDevice, meshDevice.shape(), shard2dConfig),
+        meshDevice);
   }
 }