From be02fbc21d021c5126c843519911571e8ca8a129 Mon Sep 17 00:00:00 2001 From: "Wang, Yi A" Date: Wed, 22 Oct 2025 10:44:56 +0000 Subject: [PATCH] do not use ONEDNN_XPU_INCLUDE_DIR since it's only needed for torch2_7. Signed-off-by: Wang, Yi A --- build2cmake/src/templates/xpu/torch-extension.cmake | 1 - lib/torch-extension/arch.nix | 3 --- 2 files changed, 4 deletions(-) diff --git a/build2cmake/src/templates/xpu/torch-extension.cmake b/build2cmake/src/templates/xpu/torch-extension.cmake index d2a95e47..a359f8c7 100644 --- a/build2cmake/src/templates/xpu/torch-extension.cmake +++ b/build2cmake/src/templates/xpu/torch-extension.cmake @@ -1,7 +1,6 @@ define_gpu_extension_target( {{ ops_name }} DESTINATION {{ ops_name }} - INCLUDE_DIRECTORIES "${ONEDNN_XPU_INCLUDE_DIR}" LANGUAGE ${GPU_LANG} SOURCES ${SRC} COMPILE_FLAGS ${sycl_flags} diff --git a/lib/torch-extension/arch.nix b/lib/torch-extension/arch.nix index f5baaccf..fb06d0d1 100644 --- a/lib/torch-extension/arch.nix +++ b/lib/torch-extension/arch.nix @@ -176,9 +176,6 @@ stdenv.mkDerivation (prevAttrs: { (lib.cmakeFeature "CMAKE_HIP_COMPILER_ROCM_ROOT" "${clr}") (lib.cmakeFeature "HIP_ROOT_DIR" "${clr}") ] - ++ lib.optionals xpuSupport [ - (lib.cmakeFeature "ONEDNN_XPU_INCLUDE_DIR" "${onednn-xpu}/include") - ] ++ lib.optionals stdenv.hostPlatform.isDarwin [ # Use host compiler for Metal. Not included in the redistributable SDK. (lib.cmakeFeature "METAL_COMPILER" "${xcrunHost}/bin/xcrunHost")