From d89deae9e0fa57445ae0270b8bb1a4b56ac28b0b Mon Sep 17 00:00:00 2001 From: Pei Yang Date: Mon, 19 Oct 2020 20:32:19 +0800 Subject: [PATCH] reduce trt warning message (#28011) (#28075) --- paddle/fluid/platform/dynload/tensorrt.cc | 22 +++------------------- paddle/fluid/platform/dynload/tensorrt.h | 2 ++ 2 files changed, 5 insertions(+), 19 deletions(-) diff --git a/paddle/fluid/platform/dynload/tensorrt.cc b/paddle/fluid/platform/dynload/tensorrt.cc index c9c3a9456b7..b7b8a749d2a 100644 --- a/paddle/fluid/platform/dynload/tensorrt.cc +++ b/paddle/fluid/platform/dynload/tensorrt.cc @@ -45,25 +45,9 @@ void* GetTensorRtHandle() { if (nullptr == dso_handle) { auto error_msg = - "TensorRT dynamic library (%s) that Paddle depends on is not " - "configured correctly. (error code is %s)\n" - " Suggestions:\n" - " 1. Check if TensorRT " - "is installed correctly and its version is matched with paddlepaddle " - "you installed.\n" - " 2. Configure TensorRT dynamic library environment variables as " - "follows:\n" - " - Linux: set LD_LIBRARY_PATH by `export LD_LIBRARY_PATH=...`\n" - " - Windows: set PATH by `set PATH=XXX;%PATH%`\n" - " - Mac: set DYLD_LIBRARY_PATH by `export DYLD_LIBRARY_PATH=...` " - "[Note: After Mac OS 10.11, using the DYLD_LIBRARY_PATH is " - "impossible unless System Integrity Protection (SIP) is disabled.]"; -#if !defined(_WIN32) - auto errorno = dlerror(); -#else - auto errorno = GetLastError(); -#endif // !_WIN32 - std::cerr << string::Sprintf(error_msg, dso_name, errorno); + "You are using Paddle compiled with TensorRT, but TensorRT dynamic " + "library is not found. Ignore this if TensorRT is not needed."; + std::cerr << error_msg; } return dso_handle; diff --git a/paddle/fluid/platform/dynload/tensorrt.h b/paddle/fluid/platform/dynload/tensorrt.h index cb751071062..c6650c0b041 100644 --- a/paddle/fluid/platform/dynload/tensorrt.h +++ b/paddle/fluid/platform/dynload/tensorrt.h @@ -50,6 +50,7 @@ extern void* tensorrt_dso_handle; }; \ extern DynLoad__##__name __name +#ifdef NV_TENSORRT_MAJOR #if (NV_TENSORRT_MAJOR >= 6) #define TENSORRT_RAND_ROUTINE_EACH(__macro) \ __macro(createInferBuilder_INTERNAL); \ @@ -62,6 +63,7 @@ extern void* tensorrt_dso_handle; #endif TENSORRT_RAND_ROUTINE_EACH(DECLARE_DYNAMIC_LOAD_TENSORRT_WRAP) +#endif } // namespace dynload } // namespace platform -- GitLab