Update cpuinfo.h: add support for ARM64EC

pytorch · Mar 4, 2024 · 9bb12d3 · 9bb12d3
1 parent aa4b216
commit 9bb12d3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/include/cpuinfo.h b/include/cpuinfo.h
@@ -18,15 +18,15 @@
 #define CPUINFO_ARCH_X86 1
 #endif
 
-#if defined(__x86_64__) || defined(__x86_64) || defined(_M_X64) || defined(_M_AMD64)
+#if defined(__x86_64__) || defined(__x86_64) || (defined(_M_X64) && !defined(_M_ARM64EC)) || (defined(_M_AMD64) && !defined(_M_ARM64EC))
 #define CPUINFO_ARCH_X86_64 1
 #endif
 
 #if defined(__arm__) || defined(_M_ARM)
 #define CPUINFO_ARCH_ARM 1
 #endif
 
-#if defined(__aarch64__) || defined(_M_ARM64)
+#if defined(__aarch64__) || defined(_M_ARM64) || defined(_M_ARM64EC)
 #define CPUINFO_ARCH_ARM64 1
 #endif