diff --git a/pkgs/development/cuda-modules/packages/cuda_cccl.nix b/pkgs/development/cuda-modules/packages/cuda_cccl.nix index c4f39f875dd9..ee0839f6aa5b 100644 --- a/pkgs/development/cuda-modules/packages/cuda_cccl.nix +++ b/pkgs/development/cuda-modules/packages/cuda_cccl.nix @@ -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 = '' diff --git a/pkgs/development/python-modules/torch/source/default.nix b/pkgs/development/python-modules/torch/source/default.nix index ed985066ba62..00e4c93d9c80 100644 --- a/pkgs/development/python-modules/torch/source/default.nix +++ b/pkgs/development/python-modules/torch/source/default.nix @@ -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);