Skip to content

update plugin creator name #32021

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
jiweibo merged 1 commit into from
Apr 2, 2021
Merged

update plugin creator name #32021

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions paddle/fluid/inference/tensorrt/plugin/elementwise_op_plugin.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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"; }
Expand Down Expand Up @@ -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
Expand Down
7 changes: 4 additions & 3 deletions paddle/fluid/inference/tensorrt/plugin/emb_eltwise_layernorm_plugin.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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";
}
Expand Down Expand Up @@ -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
Expand Down
6 changes: 3 additions & 3 deletions paddle/fluid/inference/tensorrt/plugin/gelu_op_plugin.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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"; }
Expand Down Expand Up @@ -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
Expand Down
6 changes: 3 additions & 3 deletions paddle/fluid/inference/tensorrt/plugin/qkv_to_context_plugin.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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"; }
Expand Down Expand Up @@ -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
Expand Down
6 changes: 3 additions & 3 deletions paddle/fluid/inference/tensorrt/plugin/skip_layernorm_op_plugin.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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"; }
Expand Down Expand Up @@ -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
Expand Down
6 changes: 3 additions & 3 deletions paddle/fluid/inference/tensorrt/plugin/slice_op_plugin.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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"; }
Expand Down Expand Up @@ -155,7 +155,7 @@ class SlicePluginV2Creator : public nvinfer1::IPluginCreator {
nvinfer1::PluginFieldCollection field_collection_;
};

REGISTER_TRT_PLUGIN_V2(SlicePluginV2Creator);
REGISTER_TRT_PLUGIN_V2(SlicePluginDynamicCreator);

#endif

Expand Down
6 changes: 3 additions & 3 deletions paddle/fluid/inference/tensorrt/plugin/split_op_plugin.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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"; }
Expand Down Expand Up @@ -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
Expand Down
6 changes: 3 additions & 3 deletions paddle/fluid/inference/tensorrt/plugin/swish_op_plugin.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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"; }
Expand Down Expand Up @@ -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
Expand Down