Expose the buildTensorRTPackage function.

2023-03-07 11:01:24 -05:00
parent c707238dc2
commit 9435abcaee
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" = [