提交 d2fc396f 编写于 作者: Z zlsh80826

use GetPluginRegistry

上级 424a2ce5
......@@ -50,7 +50,7 @@ class GeluOpConverter : public OpConverter {
#ifdef USE_NVINFER_PLUGIN
auto creator =
getPluginRegistry()->getPluginCreator("CustomGeluPluginDynamic", "1");
GetPluginRegistry()->getPluginCreator("CustomGeluPluginDynamic", "1");
assert(creator != nullptr);
int type = static_cast<int>((engine_->WithFp16() == 1)
? nvinfer1::DataType::kHALF
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册