未验证 提交 ed49b418 编写于 作者: W Wilber 提交者: GitHub

update plugin creator name (#32021)

上级 d9187869
...@@ -144,9 +144,9 @@ class ElementwisePluginDynamic : public DynamicPluginTensorRT { ...@@ -144,9 +144,9 @@ class ElementwisePluginDynamic : public DynamicPluginTensorRT {
int axis_; int axis_;
}; };
class ElementwisePluginV2Creator : public nvinfer1::IPluginCreator { class ElementwisePluginDynamicCreator : public nvinfer1::IPluginCreator {
public: public:
ElementwisePluginV2Creator() {} ElementwisePluginDynamicCreator() {}
const char* getPluginName() const override { return "elementwise_plugin"; } const char* getPluginName() const override { return "elementwise_plugin"; }
const char* getPluginVersion() const override { return "1"; } const char* getPluginVersion() const override { return "1"; }
...@@ -182,7 +182,7 @@ class ElementwisePluginV2Creator : public nvinfer1::IPluginCreator { ...@@ -182,7 +182,7 @@ class ElementwisePluginV2Creator : public nvinfer1::IPluginCreator {
std::vector<nvinfer1::PluginField> plugin_attributes_; std::vector<nvinfer1::PluginField> plugin_attributes_;
}; };
REGISTER_TRT_PLUGIN_V2(ElementwisePluginV2Creator); REGISTER_TRT_PLUGIN_V2(ElementwisePluginDynamicCreator);
#endif #endif
} // namespace plugin } // namespace plugin
......
...@@ -306,9 +306,10 @@ class EmbEltwiseLayernormPluginDynamic : public DynamicPluginTensorRT { ...@@ -306,9 +306,10 @@ class EmbEltwiseLayernormPluginDynamic : public DynamicPluginTensorRT {
} }
}; };
class EmbEltwiseLayernormPluginV2Creator : public nvinfer1::IPluginCreator { class EmbEltwiseLayernormPluginDynamicCreator
: public nvinfer1::IPluginCreator {
public: public:
EmbEltwiseLayernormPluginV2Creator() {} EmbEltwiseLayernormPluginDynamicCreator() {}
const char* getPluginName() const override { const char* getPluginName() const override {
return "fused_embedding_eltwise_layernorm_plugin"; return "fused_embedding_eltwise_layernorm_plugin";
} }
...@@ -345,7 +346,7 @@ class EmbEltwiseLayernormPluginV2Creator : public nvinfer1::IPluginCreator { ...@@ -345,7 +346,7 @@ class EmbEltwiseLayernormPluginV2Creator : public nvinfer1::IPluginCreator {
std::vector<nvinfer1::PluginField> plugin_attributes_; std::vector<nvinfer1::PluginField> plugin_attributes_;
}; };
REGISTER_TRT_PLUGIN_V2(EmbEltwiseLayernormPluginV2Creator); REGISTER_TRT_PLUGIN_V2(EmbEltwiseLayernormPluginDynamicCreator);
#endif #endif
} // namespace plugin } // namespace plugin
......
...@@ -115,9 +115,9 @@ class GeluPluginDynamic : public DynamicPluginTensorRT { ...@@ -115,9 +115,9 @@ class GeluPluginDynamic : public DynamicPluginTensorRT {
void destroy() override { delete this; } void destroy() override { delete this; }
}; };
class GeluPluginV2Creator : public nvinfer1::IPluginCreator { class GeluPluginDynamicCreator : public nvinfer1::IPluginCreator {
public: public:
GeluPluginV2Creator() {} GeluPluginDynamicCreator() {}
const char* getPluginName() const override { return "gelu_plugin"; } const char* getPluginName() const override { return "gelu_plugin"; }
const char* getPluginVersion() const override { return "1"; } const char* getPluginVersion() const override { return "1"; }
...@@ -153,7 +153,7 @@ class GeluPluginV2Creator : public nvinfer1::IPluginCreator { ...@@ -153,7 +153,7 @@ class GeluPluginV2Creator : public nvinfer1::IPluginCreator {
std::vector<nvinfer1::PluginField> plugin_attributes_; std::vector<nvinfer1::PluginField> plugin_attributes_;
}; };
REGISTER_TRT_PLUGIN_V2(GeluPluginV2Creator); REGISTER_TRT_PLUGIN_V2(GeluPluginDynamicCreator);
#endif #endif
} // namespace plugin } // namespace plugin
......
...@@ -118,9 +118,9 @@ class QkvToContextPluginDynamic : public DynamicPluginTensorRT { ...@@ -118,9 +118,9 @@ class QkvToContextPluginDynamic : public DynamicPluginTensorRT {
float scale_; float scale_;
}; };
class QkvToContextPluginV2Creator : public nvinfer1::IPluginCreator { class QkvToContextPluginDynamicCreator : public nvinfer1::IPluginCreator {
public: public:
QkvToContextPluginV2Creator() {} QkvToContextPluginDynamicCreator() {}
const char* getPluginName() const override { return "qkv_to_context_plugin"; } const char* getPluginName() const override { return "qkv_to_context_plugin"; }
const char* getPluginVersion() const override { return "1"; } const char* getPluginVersion() const override { return "1"; }
...@@ -155,7 +155,7 @@ class QkvToContextPluginV2Creator : public nvinfer1::IPluginCreator { ...@@ -155,7 +155,7 @@ class QkvToContextPluginV2Creator : public nvinfer1::IPluginCreator {
nvinfer1::PluginFieldCollection field_collection_; nvinfer1::PluginFieldCollection field_collection_;
std::vector<nvinfer1::PluginField> plugin_attributes_; std::vector<nvinfer1::PluginField> plugin_attributes_;
}; };
REGISTER_TRT_PLUGIN_V2(QkvToContextPluginV2Creator); REGISTER_TRT_PLUGIN_V2(QkvToContextPluginDynamicCreator);
#endif #endif
} // namespace plugin } // namespace plugin
......
...@@ -119,9 +119,9 @@ class SkipLayerNormPluginDynamic : public DynamicPluginTensorRT { ...@@ -119,9 +119,9 @@ class SkipLayerNormPluginDynamic : public DynamicPluginTensorRT {
float eps_; float eps_;
}; };
class SkipLayerNormPluginV2Creator : public nvinfer1::IPluginCreator { class SkipLayerNormPluginDynamicCreator : public nvinfer1::IPluginCreator {
public: public:
SkipLayerNormPluginV2Creator() {} SkipLayerNormPluginDynamicCreator() {}
const char* getPluginName() const override { return "skip_layernorm_plugin"; } const char* getPluginName() const override { return "skip_layernorm_plugin"; }
const char* getPluginVersion() const override { return "1"; } const char* getPluginVersion() const override { return "1"; }
...@@ -156,7 +156,7 @@ class SkipLayerNormPluginV2Creator : public nvinfer1::IPluginCreator { ...@@ -156,7 +156,7 @@ class SkipLayerNormPluginV2Creator : public nvinfer1::IPluginCreator {
nvinfer1::PluginFieldCollection field_collection_; nvinfer1::PluginFieldCollection field_collection_;
std::vector<nvinfer1::PluginField> plugin_attributes_; std::vector<nvinfer1::PluginField> plugin_attributes_;
}; };
REGISTER_TRT_PLUGIN_V2(SkipLayerNormPluginV2Creator); REGISTER_TRT_PLUGIN_V2(SkipLayerNormPluginDynamicCreator);
#endif #endif
} // namespace plugin } // namespace plugin
......
...@@ -121,9 +121,9 @@ class SlicePluginDynamic : public DynamicPluginTensorRT { ...@@ -121,9 +121,9 @@ class SlicePluginDynamic : public DynamicPluginTensorRT {
cudaStream_t copy_stream_; cudaStream_t copy_stream_;
}; };
class SlicePluginV2Creator : public nvinfer1::IPluginCreator { class SlicePluginDynamicCreator : public nvinfer1::IPluginCreator {
public: public:
SlicePluginV2Creator() {} SlicePluginDynamicCreator() {}
const char* getPluginName() const override { return "slice_plugin"; } const char* getPluginName() const override { return "slice_plugin"; }
const char* getPluginVersion() const override { return "1"; } const char* getPluginVersion() const override { return "1"; }
...@@ -155,7 +155,7 @@ class SlicePluginV2Creator : public nvinfer1::IPluginCreator { ...@@ -155,7 +155,7 @@ class SlicePluginV2Creator : public nvinfer1::IPluginCreator {
nvinfer1::PluginFieldCollection field_collection_; nvinfer1::PluginFieldCollection field_collection_;
}; };
REGISTER_TRT_PLUGIN_V2(SlicePluginV2Creator); REGISTER_TRT_PLUGIN_V2(SlicePluginDynamicCreator);
#endif #endif
......
...@@ -193,9 +193,9 @@ class SplitPluginDynamic : public DynamicPluginTensorRT { ...@@ -193,9 +193,9 @@ class SplitPluginDynamic : public DynamicPluginTensorRT {
std::vector<int> output_length_; std::vector<int> output_length_;
}; };
class SplitPluginV2Creator : public nvinfer1::IPluginCreator { class SplitPluginDynamicCreator : public nvinfer1::IPluginCreator {
public: public:
SplitPluginV2Creator() {} SplitPluginDynamicCreator() {}
const char* getPluginName() const override { return "split_plugin"; } const char* getPluginName() const override { return "split_plugin"; }
const char* getPluginVersion() const override { return "1"; } const char* getPluginVersion() const override { return "1"; }
...@@ -231,7 +231,7 @@ class SplitPluginV2Creator : public nvinfer1::IPluginCreator { ...@@ -231,7 +231,7 @@ class SplitPluginV2Creator : public nvinfer1::IPluginCreator {
std::vector<nvinfer1::PluginField> plugin_attributes_; std::vector<nvinfer1::PluginField> plugin_attributes_;
}; };
REGISTER_TRT_PLUGIN_V2(SplitPluginV2Creator); REGISTER_TRT_PLUGIN_V2(SplitPluginDynamicCreator);
#endif #endif
} // namespace plugin } // namespace plugin
......
...@@ -127,9 +127,9 @@ class SwishPluginDynamic : public DynamicPluginTensorRT { ...@@ -127,9 +127,9 @@ class SwishPluginDynamic : public DynamicPluginTensorRT {
float beta_; float beta_;
}; };
class SwishPluginV2Creator : public nvinfer1::IPluginCreator { class SwishPluginDynamicCreator : public nvinfer1::IPluginCreator {
public: public:
SwishPluginV2Creator() {} SwishPluginDynamicCreator() {}
const char* getPluginName() const override { return "swish_plugin"; } const char* getPluginName() const override { return "swish_plugin"; }
const char* getPluginVersion() const override { return "1"; } const char* getPluginVersion() const override { return "1"; }
...@@ -165,7 +165,7 @@ class SwishPluginV2Creator : public nvinfer1::IPluginCreator { ...@@ -165,7 +165,7 @@ class SwishPluginV2Creator : public nvinfer1::IPluginCreator {
std::vector<nvinfer1::PluginField> plugin_attributes_; std::vector<nvinfer1::PluginField> plugin_attributes_;
}; };
REGISTER_TRT_PLUGIN_V2(SwishPluginV2Creator); REGISTER_TRT_PLUGIN_V2(SwishPluginDynamicCreator);
#endif #endif
} // namespace plugin } // namespace plugin
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册