From 954ce45b81e4a4167e07dd0b584495dbc8423e2b Mon Sep 17 00:00:00 2001
From: Pavol Rusnak <pavol@rusnak.io>
Date: Sun, 13 Apr 2025 11:28:32 +0200
Subject: [PATCH] cudaPackages.nccl: 2.21.5-1 -> 2.26.2-1

---
 pkgs/development/cuda-modules/nccl/default.nix | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/pkgs/development/cuda-modules/nccl/default.nix b/pkgs/development/cuda-modules/nccl/default.nix
index 46d2fffebd06..4b9a81098172 100644
--- a/pkgs/development/cuda-modules/nccl/default.nix
+++ b/pkgs/development/cuda-modules/nccl/default.nix
@@ -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;