#14245: added false flag from s2i partial op to avoid FPE

tenstorrent · Dec 4, 2024 · 7d4dda3 · 7d4dda3
1 parent c565d87
commit 7d4dda3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...arded_partial/sharded_to_interleaved_partial/device/sharded_to_interleaved_partial_op.cpp b/...arded_partial/sharded_to_interleaved_partial/device/sharded_to_interleaved_partial_op.cpp
@@ -60,7 +60,8 @@ operation::ProgramWithCallbacks ShardedToInterleavedPartialDeviceOperation::crea
     const auto& input_tensor = input_tensors.at(0);
     auto& output_tensor = input_tensors[1];
     // Will move with sharded ops
-    return detail::sharded_to_interleaved_multi_core(input_tensor, output_tensor, this->num_slices, this->slice_index);
+    return detail::sharded_to_interleaved_multi_core(
+        input_tensor, output_tensor, false, this->num_slices, this->slice_index);
 }
 
 }  // namespace ttnn::operations::data_movement