Unverified Commit eef729c8 authored by Connor Baker's avatar Connor Baker Committed by GitHub
Browse files

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
parents b3610b80 b0052190

pkgs/development/libraries/science/math/nccl/default.nix

+7 −5
Original line number Diff line number Diff line 
{ 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;

  };

})