fix(//third_party/tensorrt): Fix include dir for library headers

Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Jul 16, 2020 · 22ed5cf · 22ed5cf
1 parent f0e0e5c
commit 22ed5cf
Showing 1 changed file with 16 additions and 4 deletions.
diff --git a/third_party/tensorrt/local/BUILD b/third_party/tensorrt/local/BUILD
@@ -134,7 +134,10 @@ cc_library(
             "include/x86_64-linux-gnu/NvOnnxConfig.h"
         ]
     }),
-    includes = ["include/"],
+    includes = select({
+        ":aarch64_linux": ["include/aarch64-linux-gnu"],
+        "//conditions:default": ["include/x86_64-linux-gnu/"],
+    }),
     visibility = ["//visibility:private"],
 )
 
@@ -169,7 +172,10 @@ cc_library(
             "include/x86_64-linux-gnu/NvOnnxParserRuntime.h",
         ]
     }),
-    includes = ["include/"],
+    includes = select({
+        ":aarch64_linux": ["include/aarch64-linux-gnu"],
+        "//conditions:default": ["include/x86_64-linux-gnu/"],
+    }),
     visibility = ["//visibility:private"],
 )
 
@@ -204,7 +210,10 @@ cc_library(
             "include/x86_64-linux-gnu/NvCaffeParser.h",
         ]
     }),
-    includes = ["include/"],
+    includes = select({
+        ":aarch64_linux": ["include/aarch64-linux-gnu"],
+        "//conditions:default": ["include/x86_64-linux-gnu/"],
+    }),
     visibility = ["//visibility:private"],
 )
 
@@ -235,7 +244,10 @@ cc_library(
         ":aarch64_linux": glob(["include/aarch64-linux-gnu/NvInferPlugin*.h"]),
         "//conditions:default": glob(["include/x86_64-linux-gnu/NvInferPlugin*.h"])
     }),
-    includes = ["include/"],
+    includes = select({
+        ":aarch64_linux": ["include/aarch64-linux-gnu"],
+        "//conditions:default": ["include/x86_64-linux-gnu/"],
+    }),
     visibility = ["//visibility:private"],
 )