From db3a21133e5d96267aa2c3d356c15ec621d71b4c Mon Sep 17 00:00:00 2001
From: Marcos Luciano <marcoslucianops@gmail.com>
Date: Wed, 27 Nov 2024 10:18:18 -0300
Subject: [PATCH] Fix upsample layer

---
 nvdsinfer_custom_impl_Yolo/layers/upsample_layer.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nvdsinfer_custom_impl_Yolo/layers/upsample_layer.cpp b/nvdsinfer_custom_impl_Yolo/layers/upsample_layer.cpp
index 4156e0b..1357146 100644
--- 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 > 0
+#if NV_TENSORRT_MAJOR >= 8 && NV_TENSORRT_MINOR > 4
   resize->setResizeMode(nvinfer1::InterpolationMode::kNEAREST);
 #else
   resize->setResizeMode(nvinfer1::ResizeMode::kNEAREST);