nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2025-02-02 02:03:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #266077 from ConnorBaker/feat/nccl-2_19_3

Browse Source 
cudaPackages.nccl: 2.18.5-1 -> 2.19.3-1; build on platforms.linux
This commit is contained in:
Connor Baker 2023-11-10 12:48:35 -05:00 committed by GitHub
commit eef729c866
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
pkgs/development/libraries/science/math/nccl/default.nix
View File

@ -1,6 +1,7 @@
{ lib
, backendStdenv
, fetchFromGitHub
, python3
, which
, autoAddOpenGLRunpathHook
, cuda_cccl
@ -17,13 +18,13 @@ let
in
backendStdenv.mkDerivation (finalAttrs: {
pname = "nccl";
version = "2.18.5-1";
version = "2.19.3-1";
src = fetchFromGitHub {
owner = "NVIDIA";
repo = finalAttrs.pname;
rev = "v${finalAttrs.version}";
hash = "sha256-vp2WitKateEt1AzSeeEvY/wM4NnUmV7XgL/gfPRUObY=";
hash = "sha256-59FlOKM5EB5Vkm4dZBRCkn+IgIcdQehE+FyZAdTCT/A=";
};
outputs = [ "out" "dev" ];
@ -32,6 +33,7 @@ backendStdenv.mkDerivation (finalAttrs: {
which
autoAddOpenGLRunpathHook
cuda_nvcc
python3
];
buildInputs = [
@ -46,7 +48,7 @@ backendStdenv.mkDerivation (finalAttrs: {
];
preConfigure = ''
patchShebangs src/collectives/device/gen_rules.sh
patchShebangs ./src/device/generate.py
makeFlagsArray+=(
"NVCC_GENCODE=${gencode}"
)
@ -77,7 +79,7 @@ backendStdenv.mkDerivation (finalAttrs: {
description = "Multi-GPU and multi-node collective communication primitives for NVIDIA GPUs";
homepage = "https://developer.nvidia.com/nccl";
license = licenses.bsd3;
platforms = [ "x86_64-linux" ];
maintainers = with maintainers; [ mdaiter orivej ];
platforms = platforms.linux;
maintainers = with maintainers; [ mdaiter orivej ] ++ teams.cuda.members;
};
})