add support for pack untilize when block rt>1

tenstorrent · Mar 13, 2024 · bc725b4 · bc725b4
1 parent 0b4d16d
commit bc725b4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llk_lib/llk_pack_untilize.h b/llk_lib/llk_pack_untilize.h
@@ -84,14 +84,14 @@ inline void _llk_pack_untilize_init_(const std::uint32_t pack_dst_format, const
 }
 
 template <std::uint32_t block_ct_dim, std::uint32_t full_ct_dim = block_ct_dim>
-inline void _llk_pack_untilize_(const std::uint32_t address, const std::uint32_t pack_dst_format,const std::uint32_t face_r_dim = FACE_R_DIM, const std::uint32_t num_faces = 4 /*not used*/) {
+inline void _llk_pack_untilize_(const std::uint32_t address, const std::uint32_t pack_dst_format,const std::uint32_t face_r_dim = FACE_R_DIM, const std::uint32_t num_faces = 4 /*not used*/, const std::uint32_t tile_dst_offset = 0) {
 
     program_packer_untilized_destination<block_ct_dim, full_ct_dim>(address, pack_dst_format);
 
     const std::uint32_t num_rows = (face_r_dim < FACE_R_DIM) ? face_r_dim : TILE_R_DIM/4;
 
     for (std::uint32_t row=0; row<num_rows; row++) {
-        TTI_SETADC(p_setadc::PAC, p_setadc::CH_0, p_setadc::SET_W, 0); // Clear tile counter
+        TT_SETADC(p_setadc::PAC, p_setadc::CH_0, p_setadc::SET_W, tile_dst_offset); // Clear tile counter
         ckernel::ckernel_template::run(instrn_buffer);
         TTI_ADDRCRXY(p_setadc::PAC, 0, 0, 1, 0, 0b0010); // Read new row in the tile
         if constexpr (block_ct_dim != full_ct_dim) {