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