Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
未验证 提交 26e2b918 编写于 作者: W Wangzheee 提交者： GitHub
浏览文件
操作

ernie: revert skip_layernorm_fp16 (#39991)

上级 1ff1c1e0
显示空白变更内容
内联 并排
Showing with 1 addition and 1 deletion
paddle/fluid/inference/tensorrt/convert/skip_layernorm.cc
浏览文件 @ 26e2b918
...@@ -105,7 +105,7 @@ class SkipLayerNormOpConverter : public OpConverter { ...@@ -105,7 +105,7 @@ class SkipLayerNormOpConverter : public OpConverter {
"in CustomSkipLayerNormPluginDynamic hidden " "in CustomSkipLayerNormPluginDynamic hidden "
"dimension should > 0")); "dimension should > 0"));
if (enable_int8) { if (enable_int8) {
type = static_cast<int>(nvinfer1::DataType::kINT8); type = static_cast<int>(nvinfer1::DataType::kHALF);
} }
const std::vector<nvinfer1::PluginField> fields{ const std::vector<nvinfer1::PluginField> fields{
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或