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

[Hardware][PowerPC] Make oneDNN dependency optional for Power (#9039)

Browse Source 
Signed-off-by: Varad Ahirwadkar <varad.ahirwadkar1@ibm.com>
This commit is contained in:
Varad Ahirwadkar 2024-10-04 22:54:42 +05:30 committed by GitHub
parent 36eecfbddb
commit e5dc713c23
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
cmake/cpu_extension.cmake
View File

@ -84,7 +84,12 @@ endif()
message(STATUS "CPU extension compile flags: ${CXX_COMPILE_FLAGS}")
list(APPEND LIBS dnnl numa)
list(APPEND LIBS numa)
# Appending the dnnl library for the AVX2 and AVX512, as it is not utilized by Power architecture.
if (AVX2_FOUND OR AVX512_FOUND)
list(APPEND LIBS dnnl)
endif()
#
# _C extension