646b892680
Some checks failed
Periodic Merges (6h) / master → staging-nixos (push) Failing after 12m50s
Periodic Merges (6h) / master → staging-next (push) Failing after 12m54s
Periodic Merges (24h) / merge-base(master,staging) → haskell-updates (push) Failing after 11m54s
Periodic Merges (6h) / staging-next → staging (push) Failing after 12m13s
Periodic Merges (24h) / staging-next-25.05 → staging-25.05 (push) Failing after 13m24s
Periodic Merges (24h) / release-25.05 → staging-next-25.05 (push) Failing after 14m28s
99 lines
2.2 KiB
Nix
99 lines
2.2 KiB
Nix
# NOTE: Though NCCL is called within the cudaPackages package set, we avoid passing in
|
|
# the names of dependencies from that package set directly to avoid evaluation errors
|
|
# in the case redistributable packages are not available.
|
|
{
|
|
lib,
|
|
fetchFromGitHub,
|
|
python3,
|
|
which,
|
|
autoAddDriverRunpath,
|
|
cudaPackages,
|
|
# passthru.updateScript
|
|
gitUpdater,
|
|
}:
|
|
let
|
|
inherit (cudaPackages)
|
|
backendStdenv
|
|
cuda_cccl
|
|
cuda_cudart
|
|
cuda_nvcc
|
|
cudaAtLeast
|
|
flags
|
|
;
|
|
version = "2.27.6-1";
|
|
hash = "sha256-/BiLSZaBbVIqOfd8nQlgUJub0YR3SR4B93x2vZpkeiU=";
|
|
in
|
|
backendStdenv.mkDerivation (finalAttrs: {
|
|
pname = "nccl";
|
|
version = version;
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "NVIDIA";
|
|
repo = "nccl";
|
|
rev = "v${finalAttrs.version}";
|
|
hash = hash;
|
|
};
|
|
|
|
__structuredAttrs = true;
|
|
strictDeps = true;
|
|
|
|
outputs = [
|
|
"out"
|
|
"dev"
|
|
];
|
|
|
|
nativeBuildInputs = [
|
|
which
|
|
autoAddDriverRunpath
|
|
python3
|
|
cuda_nvcc
|
|
];
|
|
|
|
buildInputs = [
|
|
cuda_nvcc # crt/host_config.h
|
|
cuda_cudart
|
|
cuda_cccl
|
|
];
|
|
|
|
env.NIX_CFLAGS_COMPILE = toString [ "-Wno-unused-function" ];
|
|
|
|
postPatch = ''
|
|
patchShebangs ./src/device/generate.py
|
|
patchShebangs ./src/device/symmetric/generate.py
|
|
'';
|
|
|
|
makeFlags = [
|
|
"PREFIX=$(out)"
|
|
"NVCC_GENCODE=${flags.gencodeString}"
|
|
"CUDA_HOME=${cuda_nvcc}"
|
|
"CUDA_LIB=${lib.getLib cuda_cudart}/lib"
|
|
"CUDA_INC=${lib.getDev cuda_cudart}/include"
|
|
];
|
|
|
|
enableParallelBuilding = true;
|
|
|
|
postFixup = ''
|
|
moveToOutput lib/libnccl_static.a $dev
|
|
'';
|
|
|
|
passthru.updateScript = gitUpdater {
|
|
inherit (finalAttrs) pname version;
|
|
rev-prefix = "v";
|
|
};
|
|
|
|
meta = with lib; {
|
|
description = "Multi-GPU and multi-node collective communication primitives for NVIDIA GPUs";
|
|
homepage = "https://developer.nvidia.com/nccl";
|
|
license = licenses.bsd3;
|
|
platforms = platforms.linux;
|
|
# NCCL is not supported on Jetson, because it does not use NVLink or PCI-e for inter-GPU communication.
|
|
# https://forums.developer.nvidia.com/t/can-jetson-orin-support-nccl/232845/9
|
|
badPlatforms = lib.optionals flags.isJetsonBuild [ "aarch64-linux" ];
|
|
maintainers = with maintainers; [
|
|
mdaiter
|
|
orivej
|
|
];
|
|
teams = [ teams.cuda ];
|
|
};
|
|
})
|