diff --git a/paddle/fluid/inference/tensorrt/convert/skip_layernorm.cc b/paddle/fluid/inference/tensorrt/convert/skip_layernorm.cc index 71c4348685e1b01e158aa298c48953fc3a354cec..753cd70727643d660bb1ffd3607706613f595c78 100644 --- 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(nvinfer1::DataType::kINT8); + type = static_cast(nvinfer1::DataType::kHALF); } const std::vector fields{