Add DeepStream 5.1 support

2023-05-31 15:58:17 -03:00
parent b2c4bee8dc
commit 3f14b0d95d
14 changed files with 236 additions and 4 deletions
--- a/nvdsinfer_custom_impl_Yolo/yoloPlugins.h
+++ b/nvdsinfer_custom_impl_Yolo/yoloPlugins.h
@@ -71,8 +71,12 @@ class YoloLayer : public nvinfer1::IPluginV2 {
      return maxBatchSize * sizeof(int);
    }

+#ifdef LEGACY
+    int enqueue(int batchSize, const void* const* inputs, void** outputs, void* workspace, cudaStream_t stream) override;
+#else
    int32_t enqueue(int batchSize, void const* const* inputs, void* const* outputs, void* workspace, cudaStream_t stream)
        noexcept override;
+#endif

    size_t getSerializationSize() const noexcept override;