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

This commit is contained in:
Sheng Yao
2025-03-03 09:35:01 +08:00
committed by GitHub
parent cf069aa8aa
commit 09e56f9262

View File

@ -2,6 +2,10 @@
#include <torch/all.h>
#include <cmath>
#if defined(__APPLE__)
#include "omp.h"
#endif
namespace vec_op {
#ifdef ARM_BF16_SUPPORT