From 1b96a01ec39b4f10b989e47b5afdb4a0b57ef1af Mon Sep 17 00:00:00 2001
From: Marcos Luciano <marcoslucianops@gmail.com>
Date: Wed, 13 Nov 2024 10:10:12 -0300
Subject: [PATCH] Fix DeepStream 5 and 6

---
 nvdsinfer_custom_impl_Yolo/yolo.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/nvdsinfer_custom_impl_Yolo/yolo.cpp b/nvdsinfer_custom_impl_Yolo/yolo.cpp
index 87c0446..1016e48 100644
--- a/nvdsinfer_custom_impl_Yolo/yolo.cpp
+++ b/nvdsinfer_custom_impl_Yolo/yolo.cpp
@@ -193,7 +193,12 @@ Yolo::createEngine(nvinfer1::IBuilder* builder)
   config->setProfilingVerbosity(nvinfer1::ProfilingVerbosity::kDETAILED);
 #endif
 
+#if NV_TENSORRT_MAJOR >= 8 && NV_TENSORRT_MINOR > 0
   nvinfer1::IRuntime* runtime = nvinfer1::createInferRuntime(*builder->getLogger());
+#else
+  nvinfer1::IRuntime* runtime = nvinfer1::createInferRuntime(logger);
+#endif
+
   assert(runtime);
 
   nvinfer1::IHostMemory* serializedEngine = builder->buildSerializedNetwork(*network, *config);