pkgs/development/compilers/cudatoolkit/extension.nix


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

final: prev: let
  ### Cuda Toolkit

  # Function to build the class cudatoolkit package
  buildCudaToolkitPackage = final.callPackage ./common.nix;

  # Version info for the classic cudatoolkit packages that contain everything that is in redist.
  cudatoolkitVersions = final.lib.importTOML ./versions.toml;

  ### Add classic cudatoolkit package
  cudatoolkit = buildCudaToolkitPackage ((attrs: attrs // { gcc = prev.pkgs.${attrs.gcc}; }) cudatoolkitVersions.${final.cudaVersion});

  cudaFlags = final.callPackage ./flags.nix {};

in {
  inherit cudatoolkit cudaFlags;
}