diff --git a/tt_metal/hw/ckernels/blackhole/metal/llk_api/llk_unpack_A_api.h b/tt_metal/hw/ckernels/blackhole/metal/llk_api/llk_unpack_A_api.h index 4baaf0573a8..cbaa115012c 100644 --- a/tt_metal/hw/ckernels/blackhole/metal/llk_api/llk_unpack_A_api.h +++ b/tt_metal/hw/ckernels/blackhole/metal/llk_api/llk_unpack_A_api.h @@ -10,14 +10,17 @@ * LLK UNPACK A *************************************************************************/ -template +template < + bool is_fp32_dest_acc_en = false, + StochRndType stoch_rnd_mode = StochRndType::None, + bool disable_src_zero_flag = false> inline void llk_unpack_A_hw_configure( const llk_unpack_A_params_t* unpack_A_params, const int within_face_16x16_transpose = 0) { const uint32_t unpA_operand_id = get_operand_id(unpack_A_params->unpA_operand); const uint32_t unpA_num_faces = get_operand_num_faces(unpA_operand_id); const uint32_t unpA_face_r_dim = get_operand_face_r_dim(unpA_operand_id); - _llk_unpack_A_hw_configure_( + _llk_unpack_A_hw_configure_( unpack_src_format[unpA_operand_id], unpack_dst_format[unpA_operand_id], unpA_face_r_dim, @@ -25,11 +28,15 @@ inline void llk_unpack_A_hw_configure( unpA_num_faces); } -template +template < + bool is_fp32_dest_acc_en = false, + StochRndType stoch_rnd_mode = StochRndType::None, + bool disable_src_zero_flag = false> inline void llk_unpack_A_hw_configure_disaggregated( const std::uint32_t unpA_operand, const int within_face_16x16_transpose = 0) { const llk_unpack_A_params_t unpack_A_params = {.unpA_operand = unpA_operand}; - llk_unpack_A_hw_configure(&unpack_A_params, within_face_16x16_transpose); + llk_unpack_A_hw_configure( + &unpack_A_params, within_face_16x16_transpose); } template <