cudaPackages.cuda_cccl: patch to fix torch build on cuda>=13.0 (#501079)
This commit is contained in:
@@ -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);
|
||||
|
||||
Reference in New Issue
Block a user