Make oneDNN dependency optional for Power

Signed-off-by: Varad Ahirwadkar <varad.ahirwadkar1@ibm.com>
vllm-project · Oct 2, 2024 · 4640a61 · 4640a61
1 parent 35bd215
commit 4640a61
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)
+if (POWER9_FOUND OR POWER10_FOUND)
+    message(WARNING "Skipping oneDNN for PowerPC.")
+    list(APPEND LIBS numa)
+else()
+    list(APPEND LIBS dnnl numa)
+endif()
 
 #
 # _C extension