nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2025-04-16 21:18:34 +00: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
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,