rearrange parameters

tenstorrent · Jan 16, 2025 · 9b54011 · 9b54011
1 parent ee91a63
commit 9b54011
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
@@ -17,8 +17,9 @@ void FullToShardShape(const ::ttnn::Tensor &input, ::ttnn::Tensor &out,
                       const std::vector<int64_t> &shardShape) {
   if (shardType == ::tt::target::MeshShardType::Replicate) {
     out = ::ttnn::distributed::distribute_tensor(
-        input, meshDevice,
-        *::ttnn::distributed::replicate_tensor_to_mesh_mapper(meshDevice));
+        input,
+        *::ttnn::distributed::replicate_tensor_to_mesh_mapper(meshDevice),
+        meshDevice);
   } else {
     LOG_ASSERT(
         input.get_shape().rank() > 1,