colin/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

torch: enable nccl on rocm

Browse Source
This commit is contained in:
Yaroslav Bolyukin 2024-02-28 15:09:12 +01:00
parent 1cd1a6e4ab
commit c365323b6a
No known key found for this signature in database
GPG Key ID: 881DF5A9179360DC
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pkgs/development/python-modules/torch/default.nix
View File

@ -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,