20231088/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Sheng Yao 2025-03-03 09:35:01 +08:00 committed by GitHub
parent cf069aa8aa
commit 09e56f9262
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
csrc/cpu/cpu_types_arm.hpp
View File

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