check avx512bf15

Signed-off-by: jiqing-feng <[email protected]>

Author: jiqing-feng

csrc/cpu_ops.h

@@ -298,6 +298,7 @@ static inline bool has_avx512f() {
     return v;
 }
 
+#if defined(__AVX512BF16__)
 static inline bool has_avx512bf16() {
     static bool v = [] {
         int info[4];
@@ -306,18 +307,21 @@ static inline bool has_avx512bf16() {
     }();
     return v;
 }
+#endif
 #else
 static inline bool has_avx512f() {
     static const bool supported_avx512f = __builtin_cpu_supports("avx512f");
     return supported_avx512f;
 }
 
+#if defined(__AVX512BF16__)
 static inline bool has_avx512bf16() {
     static const bool supported_avx512bf16 = __builtin_cpu_supports("avx512bf16");
     return supported_avx512bf16;
 }
 #endif
 #endif
+#endif
 
 #if defined(__AVX512F__) && defined(__AVX512BF16__)
 template <typename T, int DATA_TYPE>

csrc/pythonInterface.cpp

@@ -909,7 +909,8 @@ void gemv_4bit_inference_cpu_nf4_bf16(
 #endif
 #if defined(__AVX512F__)
 bool has_avx512f_cpu() { return has_avx512f(); }
-
+#if defined(__AVX512BF16__)
 bool has_avx512bf16_cpu() { return has_avx512bf16(); }
 #endif
+#endif
 }