Merge pull request #298609 from mschwaig/fix-torch-with-rocm

python3Packages.torchWithRocm: fix build
2024-03-30 13:34:59 +01:00
parent ef3bb68717 1c92ae9614
commit ae32f552b4
1 changed files with 3 additions and 1 deletions
@@ -52,7 +52,7 @@

  # ROCm dependencies
  rocmSupport ? config.rocmSupport,
-  rocmPackages,
+  rocmPackages_5,
  gpuTargets ? [ ]
 }:

@@ -60,6 +60,8 @@ let
  inherit (lib) attrsets lists strings trivial;
  inherit (cudaPackages) cudaFlags cudnn nccl;

+  rocmPackages = rocmPackages_5;
+
  setBool = v: if v then "1" else "0";

  # https://github.com/pytorch/pytorch/blob/v2.0.1/torch/utils/cpp_extension.py#L1744