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

09e56f92 · Sheng Yao · GitHub · cf069aa8 · 09e56f92
Unverified Commit 09e56f92 authored Mar 03, 2025 by Sheng Yao Committed by GitHub Mar 02, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

csrc/cpu/cpu_types_arm.hpp csrc/cpu/cpu_types_arm.hpp +4 -0

No files found.
--- 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