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

update plugin creator name (#32021)

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