swendel/nixpkgs
1
0
Fork 0
mirror of https://github.com/SebastianWendel/nixpkgs.git synced 2024-09-20 04:19:00 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #292097 from CertainLach/rocm-enable-nccl

Browse source 
torch: enable nccl on rocm
This commit is contained in:
Yt 2024-02-29 21:40:40 +00:00 committed by GitHub
parent 2109b8d932 c365323b6a
commit 063bbe0b3d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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,