Merge pull request #220007 from mikepurvis/expose-build-tensorrt-pkg

2023-04-09 01:25:40 +02:00
parent 9c34c8adba 9435abcaee
commit 7811f86bf9
1 changed files with 3 additions and 1 deletions
@@ -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" = [