mirror of
https://github.com/NixOS/nixpkgs.git
synced 2025-06-09 19:13:26 +03:00
_cuda: missed fixups
Signed-off-by: Connor Baker <ConnorBaker01@gmail.com>
This commit is contained in:
parent
209a4c88ec
commit
816a644da5
4 changed files with 9 additions and 11 deletions
|
@ -29,8 +29,6 @@ package set by [cuda-packages.nix](../../top-level/cuda-packages.nix).
|
|||
short, the Multiplex builder adds multiple versions of a single package to
|
||||
single instance of the CUDA Packages package set. It is used primarily for
|
||||
packages like `cudnn` and `cutensor`.
|
||||
- `lib`: A library of functions and data used by and for the CUDA package set.
|
||||
This library is exposed at the top-level as `pkgs.cudaLib`.
|
||||
- `modules`: Nixpkgs modules to check the shape and content of CUDA
|
||||
redistributable and feature manifests. These modules additionally use shims
|
||||
provided by some CUDA packages to allow them to re-use the
|
||||
|
|
|
@ -1,8 +1,8 @@
|
|||
# The _cuda attribute set is a fixed-point which contains the static functionality required to construct CUDA package
|
||||
# sets. For example, `_cuda.cudaData` includes information about NVIDIA's redistributables (such as the names NVIDIA
|
||||
# uses for different systems), `_cuda.cudaLib` contains utility functions like `formatCapabilities` (which generate
|
||||
# common arguments passed to NVCC and `cmakeFlags`), and `_cuda.cudaFixups` contains `callPackage`-able functions
|
||||
# which are provided to the corresponding package's `overrideAttrs` attribute to provide package-specific fixups
|
||||
# sets. For example, `_cuda.bootstrapData` includes information about NVIDIA's redistributables (such as the names
|
||||
# NVIDIA uses for different systems), `_cuda.lib` contains utility functions like `formatCapabilities` (which generate
|
||||
# common arguments passed to NVCC and `cmakeFlags`), and `_cuda.fixups` contains `callPackage`-able functions which
|
||||
# are provided to the corresponding package's `overrideAttrs` attribute to provide package-specific fixups
|
||||
# out of scope of the generic redistributable builder.
|
||||
#
|
||||
# Since this attribute set is used to construct the CUDA package sets, it must exist outside the fixed point of the
|
||||
|
|
|
@ -1,14 +1,13 @@
|
|||
{
|
||||
cudaData,
|
||||
cudaLib,
|
||||
_cuda,
|
||||
cudaNamePrefix,
|
||||
lib,
|
||||
runCommand,
|
||||
}:
|
||||
let
|
||||
inherit (builtins) deepSeq toJSON tryEval;
|
||||
inherit (cudaData) cudaCapabilityToInfo;
|
||||
inherit (cudaLib) formatCapabilities;
|
||||
inherit (_cuda.bootstrapData) cudaCapabilityToInfo;
|
||||
inherit (_cuda.lib) formatCapabilities;
|
||||
inherit (lib.asserts) assertMsg;
|
||||
in
|
||||
# When changing names or formats: pause, validate, and update the assert
|
||||
|
@ -62,6 +61,7 @@ assert
|
|||
"7.5"
|
||||
"8.6"
|
||||
];
|
||||
cudaForwardCompat = true;
|
||||
};
|
||||
actualWrapped = (tryEval (deepSeq actual actual)).value;
|
||||
in
|
||||
|
|
|
@ -14,7 +14,7 @@
|
|||
|
||||
let
|
||||
lib = import ../../lib;
|
||||
inherit (import ../development/cuda-modules/_cuda) cudaLib;
|
||||
cudaLib = (import ../development/cuda-modules/_cuda).lib;
|
||||
in
|
||||
|
||||
{
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue