From e0c94b8616cbb48d0a0af3fdf77c2f5f1000cb64 Mon Sep 17 00:00:00 2001
From: zlsh80826 <zlsh80826@gmail.com>
Date: Fri, 7 Aug 2020 10:03:03 +0800
Subject: [PATCH] init nvinfer_plugin

---
 paddle/fluid/inference/tensorrt/engine.h | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/paddle/fluid/inference/tensorrt/engine.h b/paddle/fluid/inference/tensorrt/engine.h
index fdd71b0d884..774d0fc4694 100644
--- a/paddle/fluid/inference/tensorrt/engine.h
+++ b/paddle/fluid/inference/tensorrt/engine.h
@@ -157,6 +157,8 @@ class TensorRTEngine {
                       "version should be at least 6.";
 #endif
     }
+
+    dy::initLibNvInferPlugins(&logger, "");
   }
 
   ~TensorRTEngine() {}
-- 
GitLab