Unverified Commit ae32f552 authored by Ulrik Strid's avatar Ulrik Strid Committed by GitHub
Browse files

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

python3Packages.torchWithRocm: fix build
parents ef3bb687 1c92ae96

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

+3 −1
Original line number Diff line number Diff line
@@ -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