gador/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

cudaPackages.cutlass: 3.9.2 -> 4.5.1

Browse Source 
Diff: https://github.com/NVIDIA/cutlass/compare/v3.9.2...v4.5.1
This commit is contained in:
Gaetan Lepage
2026-06-03 14:24:08 +00:00
parent 69b900d57c
commit c2b2a2e682
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/development/cuda-modules/packages/cutlass.nix
+10 -10
View File
@@ -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(";")]' \