From 76a71432a47fdadf4ea144dc90b9ae9f60f08e58 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Tue, 11 Aug 2026 23:14:14 +0000
Subject: [PATCH] python3Packages.nccl4py: 0.3.1 -> 0.4.1

Diff:
https://github.com/NVIDIA/nccl/compare/nccl4py-v0.3.1...nccl4py-v0.4.1

Changelog: https://github.com/NVIDIA/nccl/releases/tag/nccl4py-v0.4.1
---
 pkgs/development/python-modules/nccl4py/default.nix | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/pkgs/development/python-modules/nccl4py/default.nix b/pkgs/development/python-modules/nccl4py/default.nix
index d0ae3b556d8f..57914a461664 100644
--- a/pkgs/development/python-modules/nccl4py/default.nix
+++ b/pkgs/development/python-modules/nccl4py/default.nix
@@ -12,6 +12,7 @@
   # dependencies
   cuda-core,
   numpy,
+  nvidia-cutlass-dsl,
   packaging,
   pythonOlder,
   typing-extensions,
@@ -23,10 +24,9 @@
 
 buildPythonPackage (finalAttrs: {
   pname = "nccl4py";
-  # `nccl4py` is versioned independently of `nccl` and should be the
-  # same as the contents of
+  # `nccl4py` is versioned independently of `nccl` and should be the same as the contents of
   # `${cudaPackages.nccl.src}/bindings/nccl4py/nccl/_version.py`
-  version = "0.3.1";
+  version = "0.4.1";
   pyproject = true;
   __structuredAttrs = true;
 
@@ -34,7 +34,7 @@ buildPythonPackage (finalAttrs: {
     owner = "NVIDIA";
     repo = "nccl";
     tag = "nccl4py-v${finalAttrs.version}";
-    hash = "sha256-jpgCUnUWWl2oihtHibgoQRnnXsQC6vlIjVX73i3bVqw=";
+    hash = "sha256-p9z4NlccBdI0auMRzTJtK8VbAOSunLdwKKU0Wn5c6c0=";
   };
   sourceRoot = "${finalAttrs.src.name}/bindings/nccl4py";
 
@@ -102,6 +102,7 @@ buildPythonPackage (finalAttrs: {
   dependencies = [
     cuda-core
     numpy
+    nvidia-cutlass-dsl
     packaging
   ]
   ++ lib.optionals (pythonOlder "3.13") [
@@ -132,6 +133,7 @@ buildPythonPackage (finalAttrs: {
   meta = {
     description = "Python bindings for NCCL";
     homepage = "https://github.com/NVIDIA/nccl/blob/master/bindings/nccl4py/README.md";
+    changelog = "https://github.com/NVIDIA/nccl/releases/tag/${finalAttrs.src.tag}";
     # `cudaPackages.nccl` is BSD3 but the bindings are licensed under
     # Apache License 2.0
     license = lib.licenses.bsd3;