ernie: revert skip_layernorm_fp16 (#39991)

26e2b918 · Wangzheee · GitHub · 1ff1c1e0 · 26e2b918
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/inference/tensorrt/convert/skip_layernorm.cc paddle/fluid/inference/tensorrt/convert/skip_layernorm.cc +1 -1

未找到文件。
--- a/paddle/fluid/inference/tensorrt/convert/skip_layernorm.cc
+++ b/paddle/fluid/inference/tensorrt/convert/skip_layernorm.cc
@@ -105,7 +105,7 @@ class SkipLayerNormOpConverter : public OpConverter {
                              "in CustomSkipLayerNormPluginDynamic hidden "
                              "dimension should > 0"));
        if (enable_int8) {
-          type = static_cast<int>(nvinfer1::DataType::kINT8);
+          type = static_cast<int>(nvinfer1::DataType::kHALF);
        }

        const std::vector<nvinfer1::PluginField> fields{