Skip to content

Commit

Permalink
#11962: add GS reconfig api param
Browse files Browse the repository at this point in the history
  • Loading branch information
@rdjogoTT
rdjogoTT committed Sep 26, 2024
1 parent 14417b7 commit 94b90ae
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 6 deletions.
6 changes: 6 additions & 0 deletions tt_metal/hw/ckernels/grayskull/metal/llk_api/llk_math_common_api.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -52,18 +52,22 @@ inline void llk_math_debug_dump(std::uint8_t *data, std::uint32_t byte_size) { _
inline void llk_math_debug_dump_seek(std::uint8_t offset) { _llk_math_debug_dump_seek_(offset); }

//The following functions are only needed for WHB0, they call empty functions for GS
template <bool float_only = true /*not used*/>
inline void llk_math_reconfig_data_format_srca(const std::uint32_t srca_new_operand) {
_llk_math_reconfig_data_format_srca_();
}

template <bool float_only = true /*not used*/>
inline void llk_math_reconfig_data_format_srcb(const std::uint32_t srcb_new_operand) {
_llk_math_reconfig_data_format_srcb_();
}

template <bool float_only = true /*not used*/>
inline void llk_math_reconfig_data_format(const std::uint32_t srca_new_operand, const std::uint32_t srcb_new_operand) {
_llk_math_reconfig_data_format_();
}

template <bool float_only = true /*not used*/>
inline void llk_math_reconfig_data_format(
const std::uint32_t srca_old_operand,
const std::uint32_t srca_new_operand,
Expand All @@ -72,11 +76,13 @@ inline void llk_math_reconfig_data_format(
_llk_math_reconfig_data_format_();
}

template <bool float_only = true /*not used*/>
inline void llk_math_reconfig_data_format_srca(
const std::uint32_t srca_old_operand, const std::uint32_t srca_new_operand) {
_llk_math_reconfig_data_format_srca_();
}

template <bool float_only = true /*not used*/>
inline void llk_math_reconfig_data_format_srcb(
const std::uint32_t srcb_old_operand, const std::uint32_t srcb_new_operand) {
_llk_math_reconfig_data_format_srcb_();
Expand Down
12 changes: 6 additions & 6 deletions tt_metal/hw/ckernels/grayskull/metal/llk_api/llk_unpack_common_api.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -47,23 +47,23 @@ inline void llk_unpack_debug_dump(std::uint8_t *data, std::uint32_t byte_size) {

inline void llk_unpack_debug_dump_seek(std::uint8_t offset) { _llk_unpack_debug_dump_seek_(offset); }

template <bool is_tile_dim_reconfig_en = false /*not used*/>
template <bool float_only = true /*not used*/, bool is_tile_dim_reconfig_en = false /*not used*/>
inline void llk_unpack_reconfig_data_format_srca(const std::uint32_t srca_new_operand) {
const std::uint32_t srca_operand_id = get_operand_id(srca_new_operand);
_llk_unpack_reconfig_data_format_srca_impl_(
unpack_src_format[srca_operand_id],
unpack_dst_format[srca_operand_id]);
}

template <bool is_tile_dim_reconfig_en = false /*not used*/>
template <bool float_only = true /*not used*/, bool is_tile_dim_reconfig_en = false /*not used*/>
inline void llk_unpack_reconfig_data_format_srcb(const std::uint32_t srcb_new_operand) {
std::uint32_t srcb_operand_id = get_operand_id(srcb_new_operand);
_llk_unpack_reconfig_data_format_srcb_impl_(
unpack_src_format[srcb_operand_id],
unpack_dst_format[srcb_operand_id]);
}

template <bool is_tile_dim_reconfig_en = false /*not used*/>
template <bool float_only = true /*not used*/, bool is_tile_dim_reconfig_en = false /*not used*/>
inline void llk_unpack_reconfig_data_format_srca(
const std::uint32_t srca_old_operand, const std::uint32_t srca_new_operand) {
std::uint32_t old_srca_operand_id = get_operand_id(srca_old_operand);
Expand All @@ -76,7 +76,7 @@ inline void llk_unpack_reconfig_data_format_srca(
}
}

template <bool is_tile_dim_reconfig_en = false /*not used*/>
template <bool float_only = true /*not used*/, bool is_tile_dim_reconfig_en = false /*not used*/>
inline void llk_unpack_reconfig_data_format_srcb(
const std::uint32_t srcb_old_operand, const std::uint32_t srcb_new_operand) {
std::uint32_t old_srcb_operand_id = get_operand_id(srcb_old_operand);
Expand All @@ -89,14 +89,14 @@ inline void llk_unpack_reconfig_data_format_srcb(
}
}

template <bool is_tile_dim_reconfig_en = false /*not used*/>
template <bool float_only = true /*not used*/, bool is_tile_dim_reconfig_en = false /*not used*/>
inline void llk_unpack_reconfig_data_format(
const std::uint32_t srca_new_operand, const std::uint32_t srcb_new_operand) {
llk_unpack_reconfig_data_format_srca<is_tile_dim_reconfig_en>(srca_new_operand);
llk_unpack_reconfig_data_format_srcb<is_tile_dim_reconfig_en>(srcb_new_operand);
}

template <bool is_tile_dim_reconfig_en = false /*not used*/>
template <bool float_only = true /*not used*/, bool is_tile_dim_reconfig_en = false /*not used*/>
inline void llk_unpack_reconfig_data_format(
const std::uint32_t srca_old_operand,
const std::uint32_t srca_new_operand,
Expand Down

0 comments on commit 94b90ae

Please sign in to comment.