diff options
| author | Yt <raphael@megzari.com> | 2024-02-29 21:40:40 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-02-29 21:40:40 +0000 |
| commit | 063bbe0b3d410a481dd7b202eb24918f20693561 (patch) | |
| tree | 1d1d063ff5afcff47ed8795e40afd1f33aa0d14c /pkgs | |
| parent | 2109b8d932402ea019a0b5eb6c0e4547d684fac1 (diff) | |
| parent | c365323b6ae942826ce004a171fa5d19c5c07d40 (diff) | |
Merge pull request #292097 from CertainLach/rocm-enable-nccl
torch: enable nccl on rocm
Diffstat (limited to 'pkgs')
| -rw-r--r-- | pkgs/development/python-modules/torch/default.nix | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/pkgs/development/python-modules/torch/default.nix b/pkgs/development/python-modules/torch/default.nix index 388f66f40a49f..d6c51904bd9df 100644 --- a/pkgs/development/python-modules/torch/default.nix +++ b/pkgs/development/python-modules/torch/default.nix @@ -8,7 +8,7 @@ magma-hip, magma-cuda-static, # Use the system NCCL as long as we're targeting CUDA on a supported platform. - useSystemNccl ? (cudaSupport && !cudaPackages.nccl.meta.unsupported), + useSystemNccl ? (cudaSupport && !cudaPackages.nccl.meta.unsupported || rocmSupport), MPISupport ? false, mpi, buildDocs ? false, |