summary refs log tree commit diff
diff options
context:
space:
mode:
authorSandro <sandro.jaeckel@gmail.com>2023-04-09 01:25:40 +0200
committerGitHub <noreply@github.com>2023-04-09 01:25:40 +0200
commit7811f86bf906d01f9e2f6828f142fad28f70b241 (patch)
treee6a62bb5f7bc710bd827dfcbf81f66cda74c221a
parent9c34c8adba80180608794cce600b10183b048942 (diff)
parent9435abcaee6e68ba746deef96bde5846e0abe46e (diff)
downloadnixpkgs-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.nix4
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" = [