diff --git a/csrc/cpu/cpu_attn_vec.hpp b/csrc/cpu/cpu_attn_vec.hpp index 479313f0e..f51a232ba 100644 --- a/csrc/cpu/cpu_attn_vec.hpp +++ b/csrc/cpu/cpu_attn_vec.hpp @@ -53,7 +53,7 @@ class TileGemm82 { const int64_t ldb, const int64_t ldc, const int32_t block_size, const int32_t dynamic_k_size, const bool accum_c) { - static_assert(0 < M <= 8); + static_assert(0 < M && M <= 8); using load_vec_t = typename VecTypeTrait::vec_t; kv_cache_t* __restrict__ curr_b_0 = b_tile; diff --git a/csrc/cpu/cpu_attn_vec16.hpp b/csrc/cpu/cpu_attn_vec16.hpp index 7402312c0..06e4ad762 100644 --- a/csrc/cpu/cpu_attn_vec16.hpp +++ b/csrc/cpu/cpu_attn_vec16.hpp @@ -68,7 +68,7 @@ class TileGemm161 { const int64_t ldb, const int64_t ldc, const int32_t block_size, const int32_t dynamic_k_size, const bool accum_c) { - static_assert(0 < M <= 16); + static_assert(0 < M && M <= 16); using load_vec_t = typename VecTypeTrait::vec_t; kv_cache_t* __restrict__ curr_b_0 = b_tile; diff --git a/csrc/cpu/micro_gemm/cpu_micro_gemm_vec.hpp b/csrc/cpu/micro_gemm/cpu_micro_gemm_vec.hpp index bdd3e85a1..1c605a285 100644 --- a/csrc/cpu/micro_gemm/cpu_micro_gemm_vec.hpp +++ b/csrc/cpu/micro_gemm/cpu_micro_gemm_vec.hpp @@ -39,7 +39,7 @@ class TileGemm82 { template static void gemm_micro(DEFINE_CPU_MICRO_GEMM_PARAMS) { - static_assert(0 < M <= 8); + static_assert(0 < M && M <= 8); using load_vec_t = typename cpu_utils::VecTypeTrait::vec_t; scalar_t* __restrict__ curr_b_0 = b_ptr;