#0: Revert

tenstorrent · Jun 2, 2024 · 0d4e775 · 0d4e775
1 parent 354370a
commit 0d4e775
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tt_eager/tt_dnn/op_library/sharded/multi_core/sharded_op_multi_core.cpp b/tt_eager/tt_dnn/op_library/sharded/multi_core/sharded_op_multi_core.cpp
@@ -450,7 +450,6 @@ operation::ProgramWithCallbacks sharded_to_interleaved_multi_core(
     uint32_t curr_idx_w = 0;
 
     const auto cores = corerange_to_cores(all_cores, std::nullopt, rm_orientation);
-    uint32_t padded_shard_width = align(output_unit_size, ADDRESS_ALIGNMENT);
     for (const auto& core : cores) {
         if (input.get_layout() == Layout::TILE) {
             uint32_t shard_height = num_units_per_shard_height;
@@ -526,6 +525,7 @@ operation::ProgramWithCallbacks sharded_to_interleaved_multi_core(
                     }
                 }
             }
+            uint32_t padded_shard_width = align(shard_width, ADDRESS_ALIGNMENT);
             tt_metal::SetRuntimeArgs(
                 program,
                 unary_writer_kernel_id,