[Hardware][PowerPC] Make oneDNN dependency optional for Power (vllm-p…

…roject#9039) Signed-off-by: Varad Ahirwadkar <[email protected]>
KuntaiDu · Oct 4, 2024 · 4b43828 · 4b43828
1 parent 45bc14a
commit 4b43828
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/cmake/cpu_extension.cmake b/cmake/cpu_extension.cmake
@@ -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