fix(aarch64): fixes and issues for aarch64 toolchain

Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Jul 8, 2020 · 9a6cccd · 9a6cccd
1 parent 7889ebd
commit 9a6cccd
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 9 deletions.
diff --git a/WORKSPACE b/WORKSPACE
@@ -99,16 +99,12 @@ http_archive(
 #new_local_repository(
 #    name = "libtorch",
 #    path = "/usr/local/lib/python3.6/dist-packages/torch",
-#    # Use below for aarch64
-#    #path = "/home/nvidia/.local/lib/python3.6/site-packages/torch",
 #    build_file = "third_party/libtorch/BUILD"
 #)
 
 #new_local_repository(
 #    name = "libtorch_pre_cxx11_abi",
 #    path = "/usr/local/lib/python3.6/dist-packages/torch",
-#    # Use below for aarch64
-#    #path = "/home/nvidia/.local/lib/python3.6/site-packages/torch",
 #    build_file = "third_party/libtorch/BUILD"
 #)
 

diff --git a/tests/core/converters/BUILD b/tests/core/converters/BUILD
@@ -86,5 +86,3 @@ test_suite(
     ":test_stack"
   ]
 )
-
-
diff --git a/tests/util/BUILD b/tests/util/BUILD
@@ -23,6 +23,7 @@ cc_library(
         "//core/conversion",
         "//core/util:prelude",
         "//cpp/api:trtorch",
+	"@tensorrt//:nvinfer"
     ] + select({
         ":use_pre_cxx11_abi":  [
             "@libtorch_pre_cxx11_abi//:libtorch",

diff --git a/third_party/cuda/BUILD b/third_party/cuda/BUILD
@@ -12,7 +12,7 @@ cc_library(
     name = "cudart",
     srcs = select({
         ":aarch64_linux": [
-            "targets/aarch64-linux-gnu/lib/libcudart.so",
+            "targets/aarch64-linux/lib/libcudart.so",
         ],
         "//conditions:default": [
             "targets/x86_64-linux/lib/libcudart.so",
@@ -30,7 +30,7 @@ cc_library(
     name = "nvToolsExt",
     srcs = select({
         ":aarch64_linux": [
-            "targets/aarch64-linux-gnu/lib/libnvToolsExt.so.1",
+            "targets/aarch64-linux/lib/libnvToolsExt.so.1",
         ],
         "//conditions:default": [
             "targets/x86_64-linux/lib/libnvToolsExt.so.1",
@@ -42,7 +42,7 @@ cc_library(
     name = "cuda",
     srcs = select({
         ":aarch64_linux": glob([
-            "targets/aarch64-linux-gnu/lib/**/lib*.so",
+            "targets/aarch64-linux/lib/**/lib*.so",
         ]),
         "//conditions:default": glob([
             "targets/x86_64-linux/lib/**/lib*.so",