colin/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

cudaPackagesGoogle.cudnn_8_6: ensure present on all platforms

Browse Source 
...including jetson

Previously:
❯ nix eval -f . --argstr system aarch64-linux --arg config '{ cudaCapabilities = [ "6.2" ]; cudaEnableForwardCompat = false; cudaSupport = true; allowUnfree = true; }' -L cudaPackagesGoogle.cudnn_8_6.outPath
attribute ... in selection path .... not found

Now:
❯ nix eval -f . --argstr system aarch64-linux --arg config '{ cudaCapabilities = [ "6.2" ]; cudaEnableForwardCompat = false; cudaSupport = true; allowUnfree = true; }' -L cudaPackagesGoogle.cudnn_8_6.outPath
       error: Package ‘cudnn-8.6.0.163’ in ... is not available on the requested hostPlatform:
This commit is contained in:
Someone Serge 2024-01-20 00:48:14 +00:00
parent 9a33f8ce5b
commit 5cf0f04f95
No known key found for this signature in database
GPG Key ID: 7B0E3B1390D61DA4
2 changed files with 14 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pkgs/development/cuda-modules/generic-builders/manifest.nix
View File

@ -47,6 +47,8 @@ let
# The redistArch is the name of the architecture for which the redistributable is built.
# It is `"unsupported"` if the redistributable is not supported on the target platform.
redistArch = flags.getRedistArch hostPlatform.system;
sourceMatchesHost = flags.getNixSystem redistArch == stdenv.hostPlatform.system;
in
backendStdenv.mkDerivation (
finalAttrs: {
@ -136,7 +138,9 @@ backendStdenv.mkDerivation (
# badPlatformsConditions :: AttrSet Bool
# Sets `meta.badPlatforms = meta.platforms` if any of the conditions are true.
# Example: Broken on a specific architecture when some condition is met (like targeting Jetson).
badPlatformsConditions = { };
badPlatformsConditions = {
"No source" = !sourceMatchesHost;
};
# src :: Optional Derivation
src = trivial.pipe redistArch [

18
pkgs/development/cuda-modules/generic-builders/multiplex.nix
View File

@ -52,7 +52,7 @@ let
# - Package: ../modules/${pname}/releases/package.nix
# FIXME: do this at the module system level
propagatePlatforms = lib.mapAttrs (platform: subset: map (r: r // { inherit platform; }) subset);
propagatePlatforms = lib.mapAttrs (redistArch: packages: map (p: { inherit redistArch; } // p) packages);
# All releases across all platforms
# See ../modules/${pname}/releases/releases.nix
@ -67,8 +67,7 @@ let
# isSupported :: Package -> Bool
isSupported =
package:
# The `platform` attribute of the package is NVIDIA's name for a redistributable architecture.
redistArch == package.platform
redistArch == package.redistArch
&& strings.versionAtLeast cudaVersion package.minCudaVersion
&& strings.versionAtLeast package.maxCudaVersion cudaVersion;
@ -77,12 +76,15 @@ let
# Value is `"unsupported"` if the platform is not supported.
redistArch = flags.getRedistArch hostPlatform.system;
allReleases = lists.flatten (builtins.attrValues releaseSets);
preferable =
p1: p2: (isSupported p2 -> isSupported p1) && (strings.versionAtLeast p1.version p2.version);
# All the supported packages we can build for our platform.
# perSystemReleases :: List Package
perSystemReleases = lib.pipe (releaseSets.${redistArch} or [ ])
allReleases = lib.pipe releaseSets
[
(builtins.attrValues)
(lists.flatten)
(builtins.groupBy (p: lib.versions.majorMinor p.version))
(builtins.mapAttrs (_: builtins.sort preferable))
(builtins.mapAttrs (_: lib.take 1))
@ -90,8 +92,6 @@ let
(builtins.concatMap lib.trivial.id)
];
preferable =
p1: p2: (isSupported p2 -> isSupported p1) && (strings.versionAtLeast p1.version p2.version);
newest = builtins.head (builtins.sort preferable allReleases);
# A function which takes the `final` overlay and the `package` being built and returns
@ -115,7 +115,7 @@ let
buildPackage =
package:
let
shims = final.callPackage shimsFn {inherit package redistArch;};
shims = final.callPackage shimsFn {inherit package; inherit (package) redistArch; };
name = computeName package;
drv = final.callPackage ./manifest.nix {
inherit pname;
@ -127,7 +127,7 @@ let
attrsets.nameValuePair name fixedDrv;
# versionedDerivations :: AttrSet Derivation
versionedDerivations = builtins.listToAttrs (lists.map buildPackage perSystemReleases);
versionedDerivations = builtins.listToAttrs (lists.map buildPackage allReleases);
defaultDerivation = { ${pname} = (buildPackage newest).value; };
in