fix tensorrt 6 build error. test=develop (#27511)

* fix tensorrt 6 build error. test=develop

* fix. test=develop

* bug fix

* test=develop
revert-27520-disable_pr
Shibo Tao 4 years ago committed by GitHub
parent 081fb2f963
commit 8f7bb52bd2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -35,7 +35,7 @@ extern void* tensorrt_dso_handle;
#define DECLARE_DYNAMIC_LOAD_TENSORRT_WRAP(__name) \
struct DynLoad__##__name { \
template <typename... Args> \
auto operator()(Args... args) -> DECLARE_TYPE(__name, args...) { \
void* operator()(Args... args) { \
std::call_once(tensorrt_dso_flag, []() { \
tensorrt_dso_handle = paddle::platform::dynload::GetTensorRtHandle(); \
}); \
@ -44,7 +44,8 @@ extern void* tensorrt_dso_handle;
return nullptr; \
} \
using tensorrt_func = decltype(&::__name); \
return reinterpret_cast<tensorrt_func>(p_##__name)(args...); \
auto ret = reinterpret_cast<tensorrt_func>(p_##__name)(args...); \
return static_cast<void*>(ret); \
} \
}; \
extern DynLoad__##__name __name

Loading…
Cancel
Save