8e800cedaf
cudaPackages.cuda_compat: ignore missing libs provided at runtime cudaPackages.gpus: Jetson should never build by default cudaPackages.flags: don't build Jetson capabilities by default cudaPackages: re-introduce filter for pre-existing CUDA redist packages in overrides cudaPackages: only recurseIntoAttrs for the latest of each major version cudaPackages.nvccCompatabilities: use GCC 10 through CUDA 11.5 to avoid a GLIBC incompatability cudaPackages.cutensor: acquire libcublas through cudatoolkit prior to 11.4 cudaPackages.cuda_compat: mark as broken on aarch64-linux if not targeting Jetson cudaPackages.cutensor_1_4: fix build cudaPackages: adjust use of autoPatchelfIgnoreMissingDeps cudaPackages.cuda_nvprof: remove unecessary override to add addOpenGLRunpath cudaPackages: use getExe' to avoid patchelf warning about missing meta.mainProgram cudaPackages: fix evaluation with Nix 2.3 cudaPackages: fix platform detection for Jetson/non-Jetson aarch64-linux python3Packages.tensorrt: mark as broken if required packages are missing Note: evaluating the name of the derivation will fail if tensorrt is not present, which is why we wrap the value in `lib.optionalString`. cudaPackages.flags.getNixSystem: add guard based on jetsonTargets cudaPackages.cudnn: use explicit path to patchelf cudaPackages.tensorrt: use explicit path to patchelf
131 lines
4.1 KiB
Nix
131 lines
4.1 KiB
Nix
{
|
|
# callPackage-provided arguments
|
|
lib,
|
|
cudaVersion,
|
|
flags,
|
|
hostPlatform,
|
|
# Expected to be passed by the caller
|
|
mkVersionedPackageName,
|
|
# pname :: String
|
|
pname,
|
|
# releasesModule :: Path
|
|
# A path to a module which provides a `releases` attribute
|
|
releasesModule,
|
|
# shims :: Path
|
|
# A path to a module which provides a `shims` attribute
|
|
# The redistribRelease is only used in ./manifest.nix for the package version
|
|
# and the package description (which NVIDIA's manifest calls the "name").
|
|
# It's also used for fetching the source, but we override that since we can't
|
|
# re-use that portion of the functionality (different URLs, etc.).
|
|
# The featureRelease is used to populate meta.platforms (by way of looking at the attribute names)
|
|
# and to determine the outputs of the package.
|
|
# shimFn :: {package, redistArch} -> AttrSet
|
|
shimsFn ? ({package, redistArch}: throw "shimsFn must be provided"),
|
|
# fixupFn :: Path
|
|
# A path (or nix expression) to be evaluated with callPackage and then
|
|
# provided to the package's overrideAttrs function.
|
|
# It must accept at least the following arguments:
|
|
# - final
|
|
# - cudaVersion
|
|
# - mkVersionedPackageName
|
|
# - package
|
|
fixupFn ? (
|
|
{
|
|
final,
|
|
cudaVersion,
|
|
mkVersionedPackageName,
|
|
package,
|
|
...
|
|
}:
|
|
throw "fixupFn must be provided"
|
|
),
|
|
}:
|
|
let
|
|
inherit (lib)
|
|
attrsets
|
|
lists
|
|
modules
|
|
strings
|
|
;
|
|
|
|
evaluatedModules = modules.evalModules {
|
|
modules = [
|
|
../modules
|
|
releasesModule
|
|
];
|
|
};
|
|
|
|
# NOTE: Important types:
|
|
# - Releases: ../modules/${pname}/releases/releases.nix
|
|
# - Package: ../modules/${pname}/releases/package.nix
|
|
|
|
# All releases across all platforms
|
|
# See ../modules/${pname}/releases/releases.nix
|
|
allReleases = evaluatedModules.config.${pname}.releases;
|
|
|
|
# Compute versioned attribute name to be used in this package set
|
|
# Patch version changes should not break the build, so we only use major and minor
|
|
# computeName :: Package -> String
|
|
computeName = {version, ...}: mkVersionedPackageName pname version;
|
|
|
|
# Check whether a package supports our CUDA version
|
|
# isSupported :: Package -> Bool
|
|
isSupported =
|
|
package:
|
|
strings.versionAtLeast cudaVersion package.minCudaVersion
|
|
&& strings.versionAtLeast package.maxCudaVersion cudaVersion;
|
|
|
|
# Get all of the packages for our given platform.
|
|
redistArch = flags.getRedistArch hostPlatform.system;
|
|
|
|
# All the supported packages we can build for our platform.
|
|
# supportedPackages :: List (AttrSet Packages)
|
|
supportedPackages = builtins.filter isSupported (allReleases.${redistArch} or []);
|
|
|
|
# newestToOldestSupportedPackage :: List (AttrSet Packages)
|
|
newestToOldestSupportedPackage = lists.reverseList supportedPackages;
|
|
|
|
nameOfNewest = computeName (builtins.head newestToOldestSupportedPackage);
|
|
|
|
# A function which takes the `final` overlay and the `package` being built and returns
|
|
# a function to be consumed via `overrideAttrs`.
|
|
overrideAttrsFixupFn =
|
|
final: package:
|
|
final.callPackage fixupFn {
|
|
inherit
|
|
final
|
|
cudaVersion
|
|
mkVersionedPackageName
|
|
package
|
|
;
|
|
};
|
|
|
|
extension =
|
|
final: _:
|
|
let
|
|
# Builds our package into derivation and wraps it in a nameValuePair, where the name is the versioned name
|
|
# of the package.
|
|
buildPackage =
|
|
package:
|
|
let
|
|
shims = final.callPackage shimsFn {inherit package redistArch;};
|
|
name = computeName package;
|
|
drv = final.callPackage ./manifest.nix {
|
|
inherit pname;
|
|
redistName = pname;
|
|
inherit (shims) redistribRelease featureRelease;
|
|
};
|
|
fixedDrv = drv.overrideAttrs (overrideAttrsFixupFn final package);
|
|
in
|
|
attrsets.nameValuePair name fixedDrv;
|
|
|
|
# versionedDerivations :: AttrSet Derivation
|
|
versionedDerivations = builtins.listToAttrs (lists.map buildPackage newestToOldestSupportedPackage);
|
|
|
|
defaultDerivation = attrsets.optionalAttrs (versionedDerivations != {}) {
|
|
${pname} = versionedDerivations.${nameOfNewest};
|
|
};
|
|
in
|
|
versionedDerivations // defaultDerivation;
|
|
in
|
|
extension
|