dev/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Merge pull request #251856 from ConnorBaker/feat/nccl-update-script

Browse Source 
cudaPackages.nccl: 2.18.3-1 -> 2.18.5-1; add updateScript
This commit is contained in:
Connor Baker 2023-08-28 15:37:26 -04:00 committed by GitHub
commit 31ab8e1df0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -8,20 +8,22 @@
, cuda_nvcc
, cudaFlags
, cudaVersion
# passthru.updateScript
, gitUpdater
}:
let
# Output looks like "-gencode=arch=compute_86,code=sm_86 -gencode=arch=compute_86,code=compute_86"
gencode = lib.concatStringsSep " " cudaFlags.gencode;
in
backendStdenv.mkDerivation (finalAttrs: {
name = "nccl-${finalAttrs.version}-cuda-${cudaVersion}";
version = "2.18.3-1";
pname = "nccl";
version = "2.18.5-1";
src = fetchFromGitHub {
owner = "NVIDIA";
repo = "nccl";
repo = finalAttrs.pname;
rev = "v${finalAttrs.version}";
hash = "sha256-v4U4IzwiuiYFyFhxVmNOCUmkbSg/AM0QtWPve0ehVhs=";
hash = "sha256-vp2WitKateEt1AzSeeEvY/wM4NnUmV7XgL/gfPRUObY=";
};
outputs = [ "out" "dev" ];
@ -63,6 +65,12 @@ backendStdenv.mkDerivation (finalAttrs: {
env.NIX_CFLAGS_COMPILE = toString [ "-Wno-unused-function" ];
# Run the update script with: `nix-shell maintainers/scripts/update.nix --argstr package cudaPackages.nccl`
passthru.updateScript = gitUpdater {
inherit (finalAttrs) pname version;
rev-prefix = "v";
};
enableParallelBuilding = true;
meta = with lib; {