From 8a4d1cac17f14f14c7db6bf3402c10bc83faeccb Mon Sep 17 00:00:00 2001
From: luca <228618@studenti.unimore.it>
Date: Mon, 20 Jan 2020 14:51:38 +0100
Subject: [PATCH] compile with tensorrt 5

---
 src/NetworkRT.cpp | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/NetworkRT.cpp b/src/NetworkRT.cpp
index d4fdf90..f27f70d 100644
--- a/src/NetworkRT.cpp
+++ b/src/NetworkRT.cpp
@@ -275,18 +275,18 @@ ILayer* NetworkRT::convert_layer(ITensor *input, Activation *l) {
     if(l->act_mode == ACTIVATION_LEAKY) {
         //std::cout<<"New plugin LEAKY\n";
         
-        /*
+#if NV_TENSORRT_MAJOR < 6                
         // plugin version
         IPlugin *plugin = new ActivationLeakyRT();
         IPluginLayer *lRT = networkRT->addPlugin(&input, 1, *plugin);
         checkNULL(lRT);
         return lRT;
-        */
-
+#else 
         IActivationLayer *lRT = networkRT->addActivation(*input, ActivationType::kLEAKY_RELU);
         lRT->setAlpha(0.1);
         checkNULL(lRT);
         return lRT;
+#endif
 
     } else if(l->act_mode == CUDNN_ACTIVATION_RELU) {
         IActivationLayer *lRT = networkRT->addActivation(*input, ActivationType::kRELU);