Make NV_TENSORRT_MAJOR comparison consistent in the code

jkjung-avt · Oct 8, 2021 · f49f1f7 · f49f1f7
1 parent 4b1e7b2
commit f49f1f7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/plugins/yolo_layer.cu b/plugins/yolo_layer.cu
@@ -257,11 +257,11 @@ namespace nvinfer1
         }
     }
 
-#if NV_TENSORRT_MAJOR == 8
+#if NV_TENSORRT_MAJOR >= 8
     int32_t YoloLayerPlugin::enqueue(int32_t batchSize, void const* const* inputs, void* const* outputs, void* workspace, cudaStream_t stream) NOEXCEPT
-#else
+#else   // NV_TENSORRT_MAJOR < 8
     int YoloLayerPlugin::enqueue(int batchSize, const void*const * inputs, void** outputs, void* workspace, cudaStream_t stream)
-#endif
+#endif  // NV_TENSORRT_MAJOR
     {
         forwardGpu((const float* const*)inputs, (float*)outputs[0], stream, batchSize);
         return 0;