TensorRT的IPluginV2IOExt 和 IPluginV2Ext 的configurePlugin() 的virtual function override问题

最新推荐文章于 2024-09-14 16:34:23 发布

原创

最新推荐文章于 2024-09-14 16:34:23 发布 · 1.8k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#TensorRT #IPluginV2IOExt #IPluginV2Ext #configurePlugin

博客讲述了在实现TensorRT自定义插件时遇到的编译警告，即YoloLayerPlugin类中重写configurePlugin函数导致的IPluginV2Ext::configurePlugin函数被隐藏。为了解决这个问题，可以通过在YoloLayerPlugin中使用using声明来明确调用基类的configurePlugin函数，从而消除编译警告。

有类似如下代码:

namespace nvinfer1
{
    class YoloLayerPlugin : public IPluginV2IOExt {
        ...
        void configurePlugin(const PluginTensorDesc* in, int nbInput, const PluginTensorDesc* out, int nbOutput) override;
        ...
    }
    ...
}

在编译时总有如下警告:

yololayer.h(85): warning: function "nvinfer1::IPluginV2Ext::configurePlugin(const nvinfer1::Dims *, int, const nvinfer1::Dims *, int, const nvinfer1::DataType *, const nvinfer1::DataType *, const __nv_bool *, const __nv_bool *, nvinfer1::PluginFormat, int)" is hidden by "nvinfer1::YoloLayerPlugin::configurePlugin" -- virtual function override intended?
yololayer.h(85): warning: function "nvinfer1::IPluginV2Ext::configurePlugin(const nvinfer1::Dims *, int, const nvinfer1::Dims *, int, const nvinfer1::DataType *, const nvinfer1::DataType *, const bool *, const bool *, nvinfer1::PluginFormat, int)" is hidden by "nvinfer1::YoloLayerPlugin::configurePlugin" -- virtual function override inten