From c2b2a2e6820d95ac2bab0db3783099dfd1d496cd Mon Sep 17 00:00:00 2001 From: Gaetan Lepage Date: Wed, 3 Jun 2026 14:24:08 +0000 Subject: [PATCH] cudaPackages.cutlass: 3.9.2 -> 4.5.1 Diff: https://github.com/NVIDIA/cutlass/compare/v3.9.2...v4.5.1 --- .../cuda-modules/packages/cutlass.nix | 20 +++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/pkgs/development/cuda-modules/packages/cutlass.nix b/pkgs/development/cuda-modules/packages/cutlass.nix index c6e0126504f0..8310b9267827 100644 --- a/pkgs/development/cuda-modules/packages/cutlass.nix +++ b/pkgs/development/cuda-modules/packages/cutlass.nix @@ -46,13 +46,13 @@ backendStdenv.mkDerivation (finalAttrs: { # NOTE: Depends on the CUDA package set, so use cudaNamePrefix. name = "${cudaNamePrefix}-${finalAttrs.pname}-${finalAttrs.version}"; pname = "cutlass"; - version = "3.9.2"; + version = "4.5.1"; src = fetchFromGitHub { owner = "NVIDIA"; repo = "cutlass"; tag = "v${finalAttrs.version}"; - hash = "sha256-teziPNA9csYvhkG5t2ht8W8x5+1YGGbHm8VKx4JoxgI="; + hash = "sha256-rW//OyY9fZjxavT5QYjLvc5oOpQhrXiDGwM2NTFuehY="; }; # TODO: As a header-only library, we should make sure we have an `include` directory or similar which is not a @@ -97,7 +97,7 @@ backendStdenv.mkDerivation (finalAttrs: { # '_CUDA_INSTALL_PATH = "${getBin cuda_nvcc}"' + '' nixLog "patching python bindings to make cuda_install_path fail" - substituteInPlace ./python/cutlass/__init__.py \ + substituteInPlace ./python/cutlass_cppgen/__init__.py \ --replace-fail \ 'def cuda_install_path():' \ ' @@ -106,15 +106,15 @@ backendStdenv.mkDerivation (finalAttrs: { ' '' # Patch the python bindings to use environment variables set by Nixpkgs. - # https://github.com/NVIDIA/cutlass/blob/e94e888df3551224738bfa505787b515eae8352f/python/cutlass/backend/compiler.py#L80 - # https://github.com/NVIDIA/cutlass/blob/e94e888df3551224738bfa505787b515eae8352f/python/cutlass/backend/compiler.py#L81 - # https://github.com/NVIDIA/cutlass/blob/e94e888df3551224738bfa505787b515eae8352f/python/cutlass/backend/compiler.py#L317 - # https://github.com/NVIDIA/cutlass/blob/e94e888df3551224738bfa505787b515eae8352f/python/cutlass/backend/compiler.py#L319 - # https://github.com/NVIDIA/cutlass/blob/e94e888df3551224738bfa505787b515eae8352f/python/cutlass/backend/compiler.py#L344 - # https://github.com/NVIDIA/cutlass/blob/e94e888df3551224738bfa505787b515eae8352f/python/cutlass/backend/compiler.py#L360 + # https://github.com/NVIDIA/cutlass/blob/2e602843e75100d0e03934efb386b3e1e35d7907/python/cutlass_cppgen/backend/compiler.py#L83 + # https://github.com/NVIDIA/cutlass/blob/2e602843e75100d0e03934efb386b3e1e35d7907/python/cutlass_cppgen/backend/compiler.py#L84 + # https://github.com/NVIDIA/cutlass/blob/2e602843e75100d0e03934efb386b3e1e35d7907/python/cutlass_cppgen/backend/compiler.py#L320 + # https://github.com/NVIDIA/cutlass/blob/2e602843e75100d0e03934efb386b3e1e35d7907/python/cutlass_cppgen/backend/compiler.py#L322 + # https://github.com/NVIDIA/cutlass/blob/2e602843e75100d0e03934efb386b3e1e35d7907/python/cutlass_cppgen/backend/compiler.py#L347 + # https://github.com/NVIDIA/cutlass/blob/2e602843e75100d0e03934efb386b3e1e35d7907/python/cutlass_cppgen/backend/compiler.py#L363 + '' nixLog "patching python bindings to use environment variables" - substituteInPlace ./python/cutlass/backend/compiler.py \ + substituteInPlace ./python/cutlass_cppgen/backend/compiler.py \ --replace-fail \ 'self.include_paths = include_paths' \ 'self.include_paths = include_paths + [root + "/include" for root in os.getenv("CUDAToolkit_ROOT").split(";")]' \