src: cmake: update condition for ukernel

Author: dzarukin

src/CMakeLists.txt

@@ -81,10 +81,11 @@ if(DNNL_EXPERIMENTAL_SPARSE)
 endif()
 
 if(DNNL_EXPERIMENTAL_UKERNEL)
-    if(DNNL_TARGET_ARCH STREQUAL "ARCH_GENERIC")
-        message(FATAL_ERROR "ukernel API does not support generic architecture.")
+    if(DNNL_TARGET_ARCH STREQUAL "X64" OR DNNL_TARGET_ARCH STREQUAL "AARCH64")
+        message(STATUS "Experimental functionality for ukernels is enabled")
+    else()
+        message(FATAL_ERROR "ukernel API isn't supported for ${DNNL_TARGET_ARCH}.")
     endif()
-    message(STATUS "Experimental functionality for ukernels is enabled")
 endif()
 
 if(DNNL_EXPERIMENTAL_PROFILING)