try to fix fb_code sleef related build fail.

pytorch · Mar 30, 2024 · 4336fb0 · 4336fb0
1 parent 4e64b4c
commit 4336fb0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/aten/src/ATen/cpu/vec/vec256/vec256_bfloat16.h b/aten/src/ATen/cpu/vec/vec256/vec256_bfloat16.h
@@ -22,9 +22,9 @@ inline namespace CPU_CAPABILITY {
 #if defined(CPU_CAPABILITY_AVX2)
 
 #ifndef SLEEF_CONST
-#if defined (__GNUC__) || defined (__clang__) || defined(__INTEL_COMPILER)
-#define SLEEF_CONST __attribute__((const))
-#elif defined(_MSC_VER)
+#if (defined(__GNUC__) || defined(__CLANG__)) && !defined(__INTEL_COMPILER)
+#define SLEEF_CONST const
+#else
 #define SLEEF_CONST
 #endif
 #define SLEEF_CONST_OLD SLEEF_CONST

diff --git a/aten/src/ATen/cpu/vec/vec512/vec512_bfloat16.h b/aten/src/ATen/cpu/vec/vec512/vec512_bfloat16.h
@@ -20,9 +20,9 @@ inline namespace CPU_CAPABILITY {
 #if defined(CPU_CAPABILITY_AVX512)
 
 #ifndef SLEEF_CONST
-#if defined (__GNUC__) || defined (__clang__) || defined(__INTEL_COMPILER)
-#define SLEEF_CONST __attribute__((const))
-#elif defined(_MSC_VER)
+#if (defined(__GNUC__) || defined(__CLANG__)) && !defined(__INTEL_COMPILER)
+#define SLEEF_CONST const
+#else
 #define SLEEF_CONST
 #endif
 #define SLEEF_CONST_OLD SLEEF_CONST