Unverified Commit 22d2b1c8 authored by Pavol Rusnak's avatar Pavol Rusnak Committed by GitHub
Browse files

cudaPackages.nccl: 2.21.5-1 -> 2.26.2-1 (#398313)

parents 11fdebdb 954ce45b
Loading
Loading
Loading
Loading
+10 −2
Original line number Diff line number Diff line
@@ -22,16 +22,24 @@ let
    cudaOlder
    cudatoolkit
    ;
  # versions 2.26+ with CUDA 11.x error with
  # fatal error: cuda/atomic: No such file or directory
  version = if cudaAtLeast "12.0" then "2.26.2-1" else "2.25.1-1";
  hash =
    if cudaAtLeast "12.0" then
      "sha256-iLEuru3gaNLcAdH4V8VIv3gjdTGjgb2/Mr5UKOh69N4="
    else
      "sha256-3snh0xdL9I5BYqdbqdl+noizJoI38mZRVOJChgEE1I8=";
in
backendStdenv.mkDerivation (finalAttrs: {
  pname = "nccl";
  version = "2.21.5-1";
  version = version;

  src = fetchFromGitHub {
    owner = "NVIDIA";
    repo = "nccl";
    rev = "v${finalAttrs.version}";
    hash = "sha256-IF2tILwW8XnzSmfn7N1CO7jXL95gUp02guIW5n1eaig=";
    hash = hash;
  };

  __structuredAttrs = true;