Make oneDNN dependency optional for Power

Signed-off-by: Varad Ahirwadkar <varad.ahirwadkar1@ibm.com>
vllm-project · Oct 3, 2024 · ea756ae · ea756ae
1 parent 35bd215
commit ea756ae
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cmake/cpu_extension.cmake b/cmake/cpu_extension.cmake
@@ -84,7 +84,11 @@ endif()
 
 message(STATUS "CPU extension compile flags: ${CXX_COMPILE_FLAGS}")
 
-list(APPEND LIBS dnnl numa)
+list(APPEND LIBS numa)
+
+if (NOT POWER9_FOUND AND NOT POWER10_FOUND)
+    list(APPEND LIBS dnnl)
+endif()
 
 #
 # _C extension