cudaPackages.cuda_cccl: patch to fix torch build on cuda>=13.0 (#501079)

This commit is contained in:
Connor Baker
2026-03-18 18:40:46 +00:00
committed by GitHub
2 changed files with 8 additions and 0 deletions
@@ -23,6 +23,13 @@ buildRedist {
rmdir -v "$PWD/include/cccl"
'';
# NVIDIA, in their wisdom, expect CCCL to be a directory inside include.
# https://github.com/NVIDIA/cutlass/blob/087c84df83d254b5fb295a7a408f1a1d554085cf/CMakeLists.txt#L773
postInstall = lib.optionalString (cudaAtLeast "13.0") ''
nixLog "creating alias for ''${!outputInclude:?}/include/cccl"
ln -srv "''${!outputInclude:?}/include" "''${!outputInclude:?}/include/cccl"
'';
meta = {
description = "Building blocks that make it easier to write safe and efficient CUDA C++ code";
longDescription = ''
@@ -248,6 +248,7 @@ let
&& !(builtins.elem cudaPackages.cudaMajorVersion [
"11"
"12"
"13"
]);
"MPI cudatoolkit does not match cudaPackages.cudatoolkit" =
MPISupport && cudaSupport && (mpi.cudatoolkit != cudaPackages.cudatoolkit);