Skip to content

Commit 1b6fe71

Browse files
authored
Do not use ONEDNN_XPU_INCLUDE_DIR, not needed for Torch > 2.7 (#273)
Signed-off-by: Wang, Yi A <[email protected]>
1 parent 9855e2c commit 1b6fe71

File tree

2 files changed

+0
-4
lines changed

2 files changed

+0
-4
lines changed

build2cmake/src/templates/xpu/torch-extension.cmake

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
define_gpu_extension_target(
22
{{ ops_name }}
33
DESTINATION {{ ops_name }}
4-
INCLUDE_DIRECTORIES "${ONEDNN_XPU_INCLUDE_DIR}"
54
LANGUAGE ${GPU_LANG}
65
SOURCES ${SRC}
76
COMPILE_FLAGS ${sycl_flags}

lib/torch-extension/arch.nix

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -176,9 +176,6 @@ stdenv.mkDerivation (prevAttrs: {
176176
(lib.cmakeFeature "CMAKE_HIP_COMPILER_ROCM_ROOT" "${clr}")
177177
(lib.cmakeFeature "HIP_ROOT_DIR" "${clr}")
178178
]
179-
++ lib.optionals xpuSupport [
180-
(lib.cmakeFeature "ONEDNN_XPU_INCLUDE_DIR" "${onednn-xpu}/include")
181-
]
182179
++ lib.optionals stdenv.hostPlatform.isDarwin [
183180
# Use host compiler for Metal. Not included in the redistributable SDK.
184181
(lib.cmakeFeature "METAL_COMPILER" "${xcrunHost}/bin/xcrunHost")

0 commit comments

Comments
 (0)