enable unpack_to_dest for metal by default, enhance to work with uint32

tenstorrent · Mar 26, 2024 · 971ace9 · 971ace9
1 parent 62065f3
commit 971ace9
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 4 deletions.
diff --git a/common/inc/cmath_common.h b/common/inc/cmath_common.h
@@ -222,8 +222,8 @@ inline void set_dest_section_base()
 }
 
 inline constexpr bool is_32bit_input(const std::uint32_t src_format, const std::uint32_t dst_format) {
-    const uint input_df = src_format;
-    const uint output_df = dst_format;
+    const uint input_df = src_format & 0xF;
+    const uint output_df = dst_format & 0xF;
     return ((input_df == (uint)DataFormat::Int32)  || (input_df == (uint)DataFormat::Float32)) &&
            ((output_df == (uint)DataFormat::Int32) || (output_df == (uint)DataFormat::Float32));
 }

diff --git a/common/inc/cunpack_common.h b/common/inc/cunpack_common.h
@@ -412,8 +412,8 @@ namespace ckernel::unpacker
    }
 
    inline constexpr bool is_32bit_input(const std::uint32_t unpack_src_format, const std::uint32_t unpack_dst_format) {
-       const uint input_df = unpack_src_format;
-       const uint output_df = unpack_dst_format;
+       const uint input_df = unpack_src_format & 0xF;
+       const uint output_df = unpack_dst_format & 0xF;
        return ((input_df == (uint)DataFormat::Int32)  || (input_df == (uint)DataFormat::Float32)) &&
               ((output_df == (uint)DataFormat::Int32) || (output_df == (uint)DataFormat::Float32));
    }

diff --git a/llk_lib/llk_defs.h b/llk_lib/llk_defs.h
@@ -97,6 +97,9 @@ enum ReluType {
     MAX_THRESHOLD_RELU,
 };
 
+constexpr bool UnpackToDestEn = true;
+constexpr bool UnpackToDestDis = false;
+
 /*
 Stochastic rounding modes:
     None: No stochastic rounding enabled, default rounding is round to nearest even.

diff --git a/llk_lib/llk_unpack_common.h b/llk_lib/llk_unpack_common.h
@@ -99,6 +99,10 @@ inline void _llk_unpack_dbg_feature_disable_(){
                                                            // workaround for bug tenstorrent/budabackend#1372
 }
 
+inline void _llk_unpack_clear_dbg_feature_disable_(){
+    reg_write(RISCV_DEBUG_REG_DBG_FEATURE_DISABLE, 0);     // Unset debug feature disable
+}
+
 inline void _llk_enable_int8_fpu_math_() {
     enable_int8_fpu_math();
 }