[Bugfix] Explicitly include "omp.h" for MacOS to avoid installation failure (#14051)

realShengYao · web-flow · commit 09e56f926209 · 2025-03-02T17:35:01.000-08:00
diff --git a/csrc/cpu/cpu_types_arm.hpp b/csrc/cpu/cpu_types_arm.hpp
@@ -2,6 +2,10 @@
 #include <torch/all.h>
 #include <cmath>
 
+#if defined(__APPLE__)
+  #include "omp.h"
+#endif
+
 namespace vec_op {
 
 #ifdef ARM_BF16_SUPPORT
