From 09e56f92620908d3cf1c3020336460f0db8beead Mon Sep 17 00:00:00 2001
From: Sheng Yao <30943636+realShengYao@users.noreply.github.com>
Date: Mon, 3 Mar 2025 09:35:01 +0800
Subject: [PATCH] [Bugfix] Explicitly include "omp.h" for MacOS to avoid
 installation failure (#14051)

---
 csrc/cpu/cpu_types_arm.hpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/csrc/cpu/cpu_types_arm.hpp b/csrc/cpu/cpu_types_arm.hpp
index 990e99f2fc06..65ffe524af73 100644
--- 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