pyrox.dev
python3Packages.torch: Remove built-in composable_kernel headers search path (#406716)
authored by Gaétan Lepage and committed by GitHub f449a6e4 8efb5202
+5
pkgs/development/python-modules/torch/source/default.nix
+5
pkgs/development/python-modules/torch/source/default.nix
···
351
351
substituteInPlace cmake/public/LoadHIP.cmake \
352
352
--replace "set(ROCM_PATH \$ENV{ROCM_PATH})" \
353
353
"set(ROCM_PATH \$ENV{ROCM_PATH})''\nset(ROCM_VERSION ${lib.concatStrings (lib.intersperse "0" (lib.splitVersion rocmPackages.clr.version))})"
354
+
355
+
# Use composable kernel as dependency, rather than built-in third-party
356
+
substituteInPlace aten/src/ATen/CMakeLists.txt \
357
+
--replace-fail "list(APPEND ATen_HIP_INCLUDE \''${CMAKE_CURRENT_SOURCE_DIR}/../../../third_party/composable_kernel/include)" "" \
358
+
--replace-fail "list(APPEND ATen_HIP_INCLUDE \''${CMAKE_CURRENT_SOURCE_DIR}/../../../third_party/composable_kernel/library/include)" ""
354
359
''
355
360
# Detection of NCCL version doesn't work particularly well when using the static binary.
356
361
+ lib.optionalString cudaSupport ''