add back dtype_zero_pt checkk

intel · Jul 10, 2024 · 696820f · 696820f
1 parent 20da116
commit 696820f
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/include/experimental/group/gemm/impl/int4_dequantize_xe.hpp b/include/experimental/group/gemm/impl/int4_dequantize_xe.hpp
@@ -101,6 +101,18 @@ class gemm_t<
       std::is_same<remove_const_t<dtype_b>, remove_const_t<int4x2>>::value ||
           std::is_same<remove_const_t<dtype_b>, remove_const_t<int4x8>>::value,
       "this is for 4bit matB ");
+  static_assert(
+      quant_info_.quant_mode == quant_mode::INT4_ASYM_FP_ZERO
+          ? std::is_same_v<
+                remove_const_t<dtype_zero_pt>,
+                remove_const_t<dtype_a>>
+          : (std::is_same_v<
+                 remove_const_t<dtype_zero_pt>,
+                 remove_const_t<int4x2>> ||
+             std::is_same_v<
+                 remove_const_t<dtype_zero_pt>,
+                 remove_const_t<int4x8>>),
+      "this is for 4bit zero_pt ");
 
   /******** set memory attribute **********/
   static constexpr mem_space mem_space_a = mem_desc_a_t::space;