remove conversion for sqrt that uses trt Scale

Author: bddppq

WORKSPACE

@@ -20,8 +20,8 @@ http_archive(
     name = "libtorch",
     build_file = "@//third_party/libtorch:BUILD",
     strip_prefix = "libtorch",
-    urls = ["https://download.pytorch.org/libtorch/cu101/libtorch-cxx11-abi-shared-with-deps-1.4.0.zip"],
-    sha256 = "f214bfde532877aa5d4e0803e51a28fa8edd97b6a44b6615f75a70352b6b542e"
+    urls = ["https://download.pytorch.org/libtorch/cu101/libtorch-shared-with-deps-1.4.0.zip"],
+    sha256 = "062d27cc36e9406771328729bd89c73205ee124cbdb0acdeac998ca3b950d73b"
 )
 
 load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
@@ -36,19 +36,19 @@ rules_pkg_dependencies()
 
 new_local_repository(
     name = "cuda",
-    path = "/usr/local/cuda-10.1/targets/x86_64-linux/",
+    path = "/usr/local/cuda-10.1",
     build_file = "@//third_party/cuda:BUILD",
 )
 
 new_local_repository(
     name = "cudnn",
-    path = "/usr/",
+    path = "/usr/local/cuda-10.1",
     build_file = "@//third_party/cudnn:BUILD"
 )
 
 new_local_repository(
    name = "tensorrt",
-   path = "/usr/",
+   path = "/usr/local/cuda-10.1/TensorRT-6.0.1.5",
    build_file = "@//third_party/tensorrt:BUILD"
 )
 

core/conversion/converters/BUILD

@@ -16,7 +16,6 @@ cc_library(
         "impl/element_wise.cpp",
         "impl/linear.cpp",
         "impl/pooling.cpp",
-        "impl/scale.cpp",
         "impl/softmax.cpp",
         "impl/unary.cpp",
     ],

core/conversion/converters/impl/scale.cpp

@@ -9,7 +9,7 @@ cc_library(
 
 cc_import(
     name = "cudnn_lib",
-    shared_library = "lib/x86_64-linux-gnu/libcudnn.so",
+    shared_library = "lib64/libcudnn.so",
     visibility = ["//visibility:private"],
 )
 

tests/core/converters/test_scale.cpp

@@ -63,7 +63,7 @@ cc_library(
 cc_library(
     name = "c10_deps",
     srcs = [
-        "lib/libgomp-753e6e92.so.1",
+        "lib/libgomp-7c85b1e2.so.1",
         "lib/libcudart-1b201d85.so.10.1",
     ]
 )
@@ -92,6 +92,6 @@ cc_library(
     name = 'caffe2_deps',
     srcs = [
         'lib/libnvToolsExt-3965bdd0.so.1',
-        'lib/libgomp-753e6e92.so.1'
+        'lib/libgomp-7c85b1e2.so.1'
     ]
 )

third_party/cudnn/BUILD

@@ -14,7 +14,7 @@ cc_library(
 
 cc_import(
     name = "nvinfer_lib",
-    shared_library = "lib/x86_64-linux-gnu/libnvinfer.so",
+    shared_library = "lib/libnvinfer.so",
     visibility = ["//visibility:private"],
 )
 
@@ -34,7 +34,7 @@ cc_library(
 
 cc_import(
     name = "nvparsers_lib",
-    shared_library = "lib/x86_64-linux-gnu/libnvparsers.so",
+    shared_library = "lib/libnvparsers.so",
     visibility = ["//visibility:private"],
 )
 
@@ -66,7 +66,7 @@ cc_library(
 
 cc_import(
     name = "nvonnxparser_lib",
-    shared_library = "lib/x86_64-linux-gnu/libnvonnxparser.so",
+    shared_library = "lib/libnvonnxparser.so",
     visibility = ["//visibility:private"],
 )
 
@@ -95,7 +95,7 @@ cc_library(
 
 cc_import(
     name = "nvonnxparser_runtime_lib",
-    shared_library = "lib/x86_64-linux-gnu/libnvonnxparser_runtime.so",
+    shared_library = "lib/libnvonnxparser_runtime.so",
     visibility = ["//visibility:public"],
 )
 
@@ -120,7 +120,7 @@ cc_library(
 
 cc_import(
     name = "nvcaffeparser_lib",
-    shared_library = "lib/x86_64-linux-gnu/libnvcaffe_parsers.so",
+    shared_library = "lib/libnvcaffe_parsers.so",
     visibility = ["//visibility:private"],
 )
 
@@ -145,7 +145,7 @@ cc_library(
 
 cc_import(
     name = "nvinferplugin_lib",
-    shared_library = "lib/x86_64-linux-gnu/libnvinfer_plugin.so",
+    shared_library = "lib/libnvinfer_plugin.so",
     visibility = ["//visibility:private"],
 )