feat(//third_party/tensorrt): Add back TensorRT static lib in a cross

platform friendly way Signed-off-by: Naren Dasan <naren@narendasan.com> Signed-off-by: Naren Dasan <narens@nvidia.com>
pytorch · Aug 6, 2020 · d3c2e7e · d3c2e7e
1 parent 8eda27d
commit d3c2e7e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/third_party/tensorrt/local/BUILD b/third_party/tensorrt/local/BUILD
@@ -67,6 +67,10 @@ cc_import(
         ":windows": "lib/nvinfer.dll",
         "//conditions:default": "lib/x86_64-linux-gnu/libnvinfer.so",
     }),
+    static_library = select({
+        ":windows": "lib/nvinfer.lib",
+        "//conditions:default": ""
+    }),
     visibility = ["//visibility:private"],
 )