[OneDNN build] Check availability of Ukernels (#239)

This commit checks availability of ukernels before including them into
target.

Signed-off-by: Dmitrii Makarenko <[email protected]>

Author: Devjiu

third_party/cpu/CMakeLists.txt

@@ -1,11 +1,18 @@
+include(CheckSymbolExists)
 # Find OneDNN ukernel library
 find_package(dnnl CONFIG)
 if (dnnl_FOUND)
-  message(STATUS "Found OneDNN/DNNL")
-  add_compile_definitions(ONEDNN_AVAILABLE)
   get_target_property(dnnl_include DNNL::dnnl INTERFACE_INCLUDE_DIRECTORIES)
   # currently used only in triton_cpu.cc and in ConvertDotToOneDNN
   include_directories(${dnnl_include})
+  set (dnnl_header ${dnnl_include}/oneapi/dnnl/dnnl_config.h)
+  check_symbol_exists(DNNL_EXPERIMENTAL_UKERNEL ${dnnl_header}  HAS_UKERNELS)
+  if (HAS_UKERNELS)
+    message(STATUS "Found OneDNN/DNNL with Ukernels support")
+    add_compile_definitions(ONEDNN_AVAILABLE)
+  else ()
+    message(STATUS "SKIPPING found OneDNN/DNNL because Ukernels support is missing")
+  endif()
 else ()
   message(STATUS "Could NOT find OneDNN/DNNL")
 endif()
@@ -40,7 +47,7 @@ endif()
 # Configure and build Triton-CPU runtime
 set(TRITON_CPU_RUNTIME_SOURCES ${CMAKE_CURRENT_SOURCE_DIR}/runtime/cpu_runtime.cpp)
 set(TRITON_CPU_RUNTIME_LIBS LLVMSupport)
-if (dnnl_FOUND)
+if (dnnl_FOUND AND HAS_UKERNELS)
   set(TRITON_CPU_RUNTIME_SOURCES ${TRITON_CPU_RUNTIME_SOURCES} ${CMAKE_CURRENT_SOURCE_DIR}/runtime/runtime_onednn.cpp)
   set(TRITON_CPU_RUNTIME_LIBS ${TRITON_CPU_RUNTIME_LIBS} DNNL::dnnl)
 endif()

third_party/cpu/runtime/runtime_onednn.cpp

@@ -3,7 +3,7 @@
 #include "oneapi/dnnl/dnnl_ukernel.hpp"
 #include "oneapi/dnnl/dnnl_ukernel_types.h"
 #if !defined(DNNL_EXPERIMENTAL_UKERNEL)
-#error "DNNL Ukerenel ismissing"
+#error "DNNL Ukerenel is missing"
 #endif
 #endif