Fix the tensorRT code to support versions prior to 6.0

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-04-06 19:54:40 +02:00
parent 455291b6dd
commit 26dcb7b8c9
2 changed files with 17 additions and 3 deletions
@@ -55,8 +55,9 @@ public:
    nvinfer1::IBuilder *builderRT;
    nvinfer1::IRuntime *runtimeRT;
    nvinfer1::INetworkDefinition *networkRT; 
+#if NV_TENSORRT_MAJOR >= 6  
    nvinfer1::IBuilderConfig *configRT;
-    
+#endif
    nvinfer1::ICudaEngine *engineRT;
    nvinfer1::IExecutionContext *contextRT;