Skip to content

Commit fcd55ce

Browse files
varad-ahirwadkarsumitd2
authored andcommitted
[Hardware][PowerPC] Make oneDNN dependency optional for Power (vllm-project#9039)
Signed-off-by: Varad Ahirwadkar <varad.ahirwadkar1@ibm.com> Signed-off-by: Sumit Dubey <sumit.dubey2@ibm.com>
1 parent b05bf2f commit fcd55ce

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

cmake/cpu_extension.cmake

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,12 @@ endif()
8484

8585
message(STATUS "CPU extension compile flags: ${CXX_COMPILE_FLAGS}")
8686

87-
list(APPEND LIBS dnnl numa)
87+
list(APPEND LIBS numa)
88+
89+
# Appending the dnnl library for the AVX2 and AVX512, as it is not utilized by Power architecture.
90+
if (AVX2_FOUND OR AVX512_FOUND)
91+
list(APPEND LIBS dnnl)
92+
endif()
8893

8994
#
9095
# _C extension

0 commit comments

Comments
 (0)