rpath fixes and BASE_IMAGE default

NVIDIA · Oct 17, 2024 · 9f12bef · 9f12bef
1 parent 560da36
commit 9f12bef
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/mlir-tensorrt/BUILD b/mlir-tensorrt/BUILD
@@ -697,7 +697,7 @@ cc_binary(
     copts = [
         "-DMLIR_TRT_TARGET_TENSORRT",
     ],
-    linkopts = ["-Wl,-rpath,./bazel-mlir-tensorrt/external/tensorrt10_x86/targets/x86_64-linux-gnu/lib"],
+    linkopts = ["-Wl,-rpath,/opt/src/mlir-tensorrt/bazel-mlir-tensorrt/external/tensorrt10_x86/targets/x86_64-linux-gnu/lib"],
     deps = [
         ":TensorRTRegistration",
         "@llvm-project//mlir:MlirOptLib",

diff --git a/mlir-tensorrt/build_tools/bazel/Dockerfile b/mlir-tensorrt/build_tools/bazel/Dockerfile
@@ -1,4 +1,4 @@
-ARG BASE_IMAGE
+ARG BASE_IMAGE=nvcr.io/nvidia/cuda:12.5.1-cudnn-devel-ubuntu22.04
 FROM ${BASE_IMAGE}
 
 # Specify user IDs