diff options
author | Sandro <sandro.jaeckel@gmail.com> | 2023-04-09 01:25:40 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-04-09 01:25:40 +0200 |
commit | 7811f86bf906d01f9e2f6828f142fad28f70b241 (patch) | |
tree | e6a62bb5f7bc710bd827dfcbf81f66cda74c221a | |
parent | 9c34c8adba80180608794cce600b10183b048942 (diff) | |
parent | 9435abcaee6e68ba746deef96bde5846e0abe46e (diff) | |
download | nixpkgs-7811f86bf906d01f9e2f6828f142fad28f70b241.tar nixpkgs-7811f86bf906d01f9e2f6828f142fad28f70b241.tar.gz nixpkgs-7811f86bf906d01f9e2f6828f142fad28f70b241.tar.bz2 nixpkgs-7811f86bf906d01f9e2f6828f142fad28f70b241.tar.lz nixpkgs-7811f86bf906d01f9e2f6828f142fad28f70b241.tar.xz nixpkgs-7811f86bf906d01f9e2f6828f142fad28f70b241.tar.zst nixpkgs-7811f86bf906d01f9e2f6828f142fad28f70b241.zip |
Merge pull request #220007 from mikepurvis/expose-build-tensorrt-pkg
-rw-r--r-- | pkgs/development/libraries/science/math/tensorrt/extension.nix | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/pkgs/development/libraries/science/math/tensorrt/extension.nix b/pkgs/development/libraries/science/math/tensorrt/extension.nix index b4f47a8969c..507dc0cfd50 100644 --- a/pkgs/development/libraries/science/math/tensorrt/extension.nix +++ b/pkgs/development/libraries/science/math/tensorrt/extension.nix @@ -27,7 +27,9 @@ final: prev: let defaultBuild = { "tensorrt" = if allBuilds ? ${computeName tensorRTDefaultVersion} then allBuilds.${computeName tensorRTDefaultVersion} else throw "tensorrt-${tensorRTDefaultVersion} does not support your cuda version ${cudaVersion}"; }; - in allBuilds // defaultBuild; + in { + inherit buildTensorRTPackage; + } // allBuilds // defaultBuild; tensorRTVersions = { "8.4.0" = [ |