#7406: pass operand id to datacopy call

tenstorrent · May 6, 2024 · f975d0b · f975d0b
1 parent 8c1b785
commit f975d0b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tt_metal/hw/ckernels/grayskull/metal/llk_api/llk_math_unary_datacopy_api.h b/tt_metal/hw/ckernels/grayskull/metal/llk_api/llk_math_unary_datacopy_api.h
@@ -15,7 +15,7 @@ template <
     BroadcastType src_b_bcast_type = BroadcastType::NONE,
     bool is_fp32_dest_acc_en = false,
     bool unpack_to_dest = false>
-inline void llk_math_eltwise_unary_datacopy(uint dst_index) {
+inline void llk_math_eltwise_unary_datacopy(uint dst_index, uint operand = 0 /* unused */) {
 
     _llk_math_eltwise_unary_datacopy_<type, src_b_bcast_type, DstSync::SyncHalf, is_fp32_dest_acc_en>(dst_index);
 }

diff --git a/tt_metal/include/compute_kernel_api/tile_move_copy.h b/tt_metal/include/compute_kernel_api/tile_move_copy.h
@@ -76,7 +76,7 @@ ALWI void copy_tile_to_dst_init_short_with_dt(uint32_t old_cbid, uint32_t new_cb
 ALWI void copy_tile(uint32_t in_cb_id, uint32_t in_tile_index, uint32_t dst_tile_index)
 {
     UNPACK(( llk_unpack_A<BroadcastType::NONE, false, EltwiseBinaryReuseDestType::NONE, UnpackToDestEn>(in_cb_id, in_tile_index)  ));
-    MATH(( llk_math_eltwise_unary_datacopy<A2D, BroadcastType::NONE, DST_ACCUM_MODE, UnpackToDestEn>(dst_tile_index)  ));
+    MATH(( llk_math_eltwise_unary_datacopy<A2D, BroadcastType::NONE, DST_ACCUM_MODE, UnpackToDestEn>(dst_tile_index, in_cb_id)  ));
 }
 
 ALWI void copy_block_matmul_partials(uint32_t in_cb_id, uint32_t start_in_tile_index, uint32_t start_dst_tile_index, uint32_t ntiles)