From 7d4dda3d91dbedf1405f4f8d2106eec99782a3df Mon Sep 17 00:00:00 2001 From: Ligang Long Date: Wed, 4 Dec 2024 06:39:54 +0000 Subject: [PATCH] #14245: added false flag from s2i partial op to avoid FPE --- .../device/sharded_to_interleaved_partial_op.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ttnn/cpp/ttnn/operations/data_movement/sharded_partial/sharded_to_interleaved_partial/device/sharded_to_interleaved_partial_op.cpp b/ttnn/cpp/ttnn/operations/data_movement/sharded_partial/sharded_to_interleaved_partial/device/sharded_to_interleaved_partial_op.cpp index 40df37dfd50..de6714ba07b 100644 --- a/ttnn/cpp/ttnn/operations/data_movement/sharded_partial/sharded_to_interleaved_partial/device/sharded_to_interleaved_partial_op.cpp +++ b/ttnn/cpp/ttnn/operations/data_movement/sharded_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