Unverified Commit f449a6e4 authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

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

parents 8efb5202 e1cea252

pkgs/development/python-modules/torch/source/default.nix

+5 −0
Original line number Diff line number Diff line
@@ -351,6 +351,11 @@ buildPythonPackage rec { 
      substituteInPlace cmake/public/LoadHIP.cmake \

        --replace "set(ROCM_PATH \$ENV{ROCM_PATH})" \

          "set(ROCM_PATH \$ENV{ROCM_PATH})''\nset(ROCM_VERSION ${lib.concatStrings (lib.intersperse "0" (lib.splitVersion rocmPackages.clr.version))})"



      # Use composable kernel as dependency, rather than built-in third-party

      substituteInPlace aten/src/ATen/CMakeLists.txt \

        --replace-fail "list(APPEND ATen_HIP_INCLUDE \''${CMAKE_CURRENT_SOURCE_DIR}/../../../third_party/composable_kernel/include)" "" \

        --replace-fail "list(APPEND ATen_HIP_INCLUDE \''${CMAKE_CURRENT_SOURCE_DIR}/../../../third_party/composable_kernel/library/include)" ""

    ''

    # Detection of NCCL version doesn't work particularly well when using the static binary.

    + lib.optionalString cudaSupport ''