Unverified Commit 6604d5b3 authored by Connor Baker's avatar Connor Baker Committed by GitHub
Browse files

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

parents bb7d6e19 6c382094

pkgs/development/cuda-modules/packages/cuda_cccl.nix

+7 −0
Original line number Diff line number Diff line
@@ -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 = ''

pkgs/development/python-modules/torch/source/default.nix

+1 −0
Original line number Diff line number Diff line
@@ -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);