pyrox.dev / nixpkgs
lol
fork atom

python3Packages.torch: Remove built-in composable_kernel headers search path (#406716)

authored by Gaétan Lepage and committed by GitHub f449a6e4 8efb5202

+5
unified split
+5
pkgs/development/python-modules/torch/source/default.nix
··· 351 substituteInPlace cmake/public/LoadHIP.cmake \ 352 --replace "set(ROCM_PATH \$ENV{ROCM_PATH})" \ 353 "set(ROCM_PATH \$ENV{ROCM_PATH})''\nset(ROCM_VERSION ${lib.concatStrings (lib.intersperse "0" (lib.splitVersion rocmPackages.clr.version))})" 354 '' 355 # Detection of NCCL version doesn't work particularly well when using the static binary. 356 + lib.optionalString cudaSupport ''
··· 351 substituteInPlace cmake/public/LoadHIP.cmake \ 352 --replace "set(ROCM_PATH \$ENV{ROCM_PATH})" \ 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)" "" 359 '' 360 # Detection of NCCL version doesn't work particularly well when using the static binary. 361 + lib.optionalString cudaSupport ''