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

Signed-off-by: Varad Ahirwadkar <varad.ahirwadkar1@ibm.com>
vllm-project · Oct 4, 2024 · e5dc713 · e5dc713
1 parent 36eecfb
commit e5dc713
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