41 lines
1.3 KiB
Nix
41 lines
1.3 KiB
Nix
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;
|
|
|
|
finalVersion = cudatoolkitVersions.${final.cudaVersion};
|
|
|
|
# Exposed as cudaPackages.backendStdenv.
|
|
# This is what nvcc uses as a backend,
|
|
# and it has to be an officially supported one (e.g. gcc11 for cuda11).
|
|
#
|
|
# It, however, propagates current stdenv's libstdc++ to avoid "GLIBCXX_* not found errors"
|
|
# when linked with other C++ libraries.
|
|
# E.g. for cudaPackages_11_8 we use gcc11 with gcc12's libstdc++
|
|
# Cf. https://github.com/NixOS/nixpkgs/pull/218265 for context
|
|
backendStdenv = final.callPackage ./stdenv.nix {
|
|
nixpkgsStdenv = prev.pkgs.stdenv;
|
|
nvccCompatibleStdenv = prev.pkgs.buildPackages."${finalVersion.gcc}Stdenv";
|
|
};
|
|
|
|
### Add classic cudatoolkit package
|
|
cudatoolkit =
|
|
let
|
|
attrs = builtins.removeAttrs finalVersion [ "gcc" ];
|
|
attrs' = attrs // { inherit backendStdenv; };
|
|
in
|
|
buildCudaToolkitPackage attrs';
|
|
|
|
cudaFlags = final.callPackage ./flags.nix {};
|
|
|
|
in
|
|
{
|
|
inherit
|
|
backendStdenv
|
|
cudatoolkit
|
|
cudaFlags;
|
|
}
|