Fix build

marcoslucianops · Dec 6, 2024 · c11c3e4 · c11c3e4
1 parent e5d994e
commit c11c3e4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/nvdsinfer_custom_impl_Yolo/layers/upsample_layer.cpp b/nvdsinfer_custom_impl_Yolo/layers/upsample_layer.cpp
@@ -25,7 +25,7 @@ upsampleLayer(int layerIdx, std::map<std::string, std::string>& block, nvinfer1:
   std::string resizeLayerName = "upsample_" + std::to_string(layerIdx);
   resize->setName(resizeLayerName.c_str());
 
-#if NV_TENSORRT_MAJOR >= 8 && NV_TENSORRT_MINOR > 4
+#if NV_TENSORRT_MAJOR > 8 || (NV_TENSORRT_MAJOR == 8 && NV_TENSORRT_MINOR > 4)
   resize->setResizeMode(nvinfer1::InterpolationMode::kNEAREST);
 #else
   resize->setResizeMode(nvinfer1::ResizeMode::kNEAREST);

diff --git a/nvdsinfer_custom_impl_Yolo/yolo.cpp b/nvdsinfer_custom_impl_Yolo/yolo.cpp
@@ -77,7 +77,7 @@ Yolo::createEngine(nvinfer1::IBuilder* builder)
 
   if (m_NetworkType == "onnx") {
 
-#if NV_TENSORRT_MAJOR >= 8 && NV_TENSORRT_MINOR > 0
+#if NV_TENSORRT_MAJOR > 8 || (NV_TENSORRT_MAJOR == 8 && NV_TENSORRT_MINOR > 0)
     parser = nvonnxparser::createParser(*network, *builder->getLogger());
 #else
     parser = nvonnxparser::createParser(*network, logger);
@@ -193,7 +193,7 @@ Yolo::createEngine(nvinfer1::IBuilder* builder)
   config->setProfilingVerbosity(nvinfer1::ProfilingVerbosity::kDETAILED);
 #endif
 
-#if NV_TENSORRT_MAJOR >= 8 && NV_TENSORRT_MINOR > 0
+#if NV_TENSORRT_MAJOR > 8 || (NV_TENSORRT_MAJOR == 8 && NV_TENSORRT_MINOR > 0)
   nvinfer1::IRuntime* runtime = nvinfer1::createInferRuntime(*builder->getLogger());
 #else
   nvinfer1::IRuntime* runtime = nvinfer1::createInferRuntime(logger);