pyrox.dev
Merge pull request #266077 from ConnorBaker/feat/nccl-2_19_3
cudaPackages.nccl: 2.18.5-1 -> 2.19.3-1; build on platforms.linux
authored by Connor Baker and committed by GitHub eef729c8 b3610b80
+7
-5
pkgs/development/libraries/science/math/nccl/default.nix
+7
-5
pkgs/development/libraries/science/math/nccl/default.nix
···
1
1
{ lib
2
2
, backendStdenv
3
3
, fetchFromGitHub
4
+
, python3
4
5
, which
5
6
, autoAddOpenGLRunpathHook
6
7
, cuda_cccl
···
17
18
in
18
19
backendStdenv.mkDerivation (finalAttrs: {
19
20
pname = "nccl";
20
-
version = "2.18.5-1";
21
+
version = "2.19.3-1";
21
22
22
23
src = fetchFromGitHub {
23
24
owner = "NVIDIA";
24
25
repo = finalAttrs.pname;
25
26
rev = "v${finalAttrs.version}";
26
-
hash = "sha256-vp2WitKateEt1AzSeeEvY/wM4NnUmV7XgL/gfPRUObY=";
27
+
hash = "sha256-59FlOKM5EB5Vkm4dZBRCkn+IgIcdQehE+FyZAdTCT/A=";
27
28
};
28
29
29
30
outputs = [ "out" "dev" ];
···
32
33
which
33
34
autoAddOpenGLRunpathHook
34
35
cuda_nvcc
36
+
python3
35
37
];
36
38
37
39
buildInputs = [
···
46
48
];
47
49
48
50
preConfigure = ''
49
-
patchShebangs src/collectives/device/gen_rules.sh
51
+
patchShebangs ./src/device/generate.py
50
52
makeFlagsArray+=(
51
53
"NVCC_GENCODE=${gencode}"
52
54
)
···
77
79
description = "Multi-GPU and multi-node collective communication primitives for NVIDIA GPUs";
78
80
homepage = "https://developer.nvidia.com/nccl";
79
81
license = licenses.bsd3;
80
-
platforms = [ "x86_64-linux" ];
81
-
maintainers = with maintainers; [ mdaiter orivej ];
82
+
platforms = platforms.linux;
83
+
maintainers = with maintainers; [ mdaiter orivej ] ++ teams.cuda.members;
82
84
};
83
85
})