tensorrt7 support for ipluginv2

2021-10-18 18:07:57 +05:30
parent 18cc6abbb6
commit 54e7af11ed
30 changed files with 313 additions and 64 deletions
@@ -29,8 +29,13 @@ namespace nvinfer1 {

        size_t getWorkspaceSize(int maxBatchSize) const NOEXCEPT override ;

+#if NV_TENSORRT_MAJOR > 7
        int enqueue(int batchSize, const void *const *inputs, void *const *outputs, void *workspace,
                            cudaStream_t stream) NOEXCEPT override ;
+#elif NV_TENSORRT_MAJOR == 7
+        int32_t enqueue (int32_t batchSize, const void *const *inputs, void **outputs, void *workspace, cudaStream_t stream) override;
+#endif
+


        size_t getSerializationSize() const NOEXCEPT override ;