diff --git a/third_party/tsl/third_party/mkl_dnn/mkldnn_v1.BUILD b/third_party/tsl/third_party/mkl_dnn/mkldnn_v1.BUILD index 214f9f7b369b66..0bdfebb2cdabe2 100644 --- a/third_party/tsl/third_party/mkl_dnn/mkldnn_v1.BUILD +++ b/third_party/tsl/third_party/mkl_dnn/mkldnn_v1.BUILD @@ -6,13 +6,16 @@ exports_files(["LICENSE"]) _CMAKE_COMMON_LIST = { "#cmakedefine DNNL_GPU_RUNTIME DNNL_RUNTIME_${DNNL_GPU_RUNTIME}": "#define DNNL_GPU_RUNTIME DNNL_RUNTIME_NONE", + "#cmakedefine DNNL_GPU_VENDOR DNNL_VENDOR_${DNNL_GPU_VENDOR}": "#define DNNL_VENDOR_NONE", "#cmakedefine DNNL_USE_RT_OBJECTS_IN_PRIMITIVE_CACHE": "#undef DNNL_USE_RT_OBJECTS_IN_PRIMITIVE_CACHE", "#cmakedefine DNNL_WITH_SYCL": "#undef DNNL_WITH_SYCL", + "#cmakedefine DNNL_SYCL_GENERIC": "#define DNNL_SYCL_GENERIC 1", "#cmakedefine DNNL_WITH_LEVEL_ZERO": "#undef DNNL_WITH_LEVEL_ZERO", "#cmakedefine DNNL_SYCL_CUDA": "#undef DNNL_SYCL_CUDA", "#cmakedefine DNNL_SYCL_HIP": "#undef DNNL_SYCL_HIP", "#cmakedefine DNNL_ENABLE_STACK_CHECKER": "#undef DNNL_ENABLE_STACK_CHECKER", "#cmakedefine ONEDNN_BUILD_GRAPH": "#define ONEDNN_BUILD_GRAPH", + "#cmakedefine DNNL_DISABLE_GPU_REF_KERNELS": "#define DNNL_DISABLE_GPU_REF_KERNELS 0", "#cmakedefine DNNL_EXPERIMENTAL_SPARSE": "#define DNNL_EXPERIMENTAL_SPARSE", "#cmakedefine DNNL_EXPERIMENTAL": "#undef DNNL_EXPERIMENTAL", "#cmakedefine01 BUILD_TRAINING": "#define BUILD_TRAINING 1", @@ -51,6 +54,7 @@ _CMAKE_COMMON_LIST = { "#cmakedefine01 BUILD_PRIMITIVE_GPU_ISA_ALL": "#define BUILD_PRIMITIVE_GPU_ISA_ALL 0", "#cmakedefine01 BUILD_GEN9": "#define BUILD_GEN9 0", "#cmakedefine01 BUILD_GEN11": "#define BUILD_GEN11 0", + "#cmakedefine01 BUILD_SDPA": "#define BUILD_SDPA 1", "#cmakedefine01 BUILD_XE2": "#define BUILD_XE2 0", "#cmakedefine01 BUILD_XELP": "#define BUILD_XELP 0", "#cmakedefine01 BUILD_XEHPG": "#define BUILD_XEHPG 0", @@ -94,13 +98,21 @@ expand_template( out = "include/oneapi/dnnl/dnnl_version.h", substitutions = { "@DNNL_VERSION_MAJOR@": "3", - "@DNNL_VERSION_MINOR@": "5", + "@DNNL_VERSION_MINOR@": "6", "@DNNL_VERSION_PATCH@": "0", - "@DNNL_VERSION_HASH@": "N/A", }, template = "include/oneapi/dnnl/dnnl_version.h.in", ) +expand_template( + name = "dnnl_version_hash_h", + out = "include/oneapi/dnnl/dnnl_version_hash.h", + substitutions = { + "@DNNL_VERSION_HASH@": "N/A", + }, + template = "include/oneapi/dnnl/dnnl_version_hash.h.in", +) + _COPTS_LIST = select({ "@xla//xla/tsl:windows": [], "//conditions:default": ["-fexceptions"], @@ -143,6 +155,7 @@ _TEXTUAL_HDRS_LIST = glob([ ]) + [ ":dnnl_config_h", ":dnnl_version_h", + ":dnnl_version_hash_h", ] # Large autogen files take too long time to compile with usual optimization @@ -186,6 +199,7 @@ cc_library( "src/cpu/aarch64/**", "src/cpu/rv64/**", "src/cpu/x64/gemm/**/*_kern_autogen.cpp", + "src/cpu/sycl/**", ], ), copts = _COPTS_LIST,