[Paddle Inference]Disable skip layernorm half (#45047)

* disable_skip_layernorm_fp16

[Paddle Inference]Disable skip layernorm half (#45047)
* disable_skip_layernorm_fp16
4805da50 · Wangzheee · GitHub · 3f49817a · 4805da50
显示空白变更内容
内联并排

Showing with 4 addition and 2 deletion

paddle/fluid/inference/tensorrt/convert/skip_layernorm.cc paddle/fluid/inference/tensorrt/convert/skip_layernorm.cc +4 -2

未找到文件。
--- a/paddle/fluid/inference/tensorrt/convert/skip_layernorm.cc
+++ b/paddle/fluid/inference/tensorrt/convert/skip_layernorm.cc
@@ -163,8 +163,10 @@ class SkipLayerNormOpConverter : public OpConverter {
      auto scale_weight = GetFp32Weight("Scale").get();

      float eps = PADDLE_GET_CONST(float, op_desc.GetAttr("epsilon"));
-      bool with_fp16 =
-          engine_->WithFp16() && !engine_->disable_trt_plugin_fp16();
+      // bool with_fp16 =
+      //     engine_->WithFp16() && !engine_->disable_trt_plugin_fp16();
+      bool with_fp16 = false;
+
      plugin::SkipLayerNormPluginDynamic* plugin =
          new plugin::SkipLayerNormPluginDynamic(
              static_cast<const float*>(bias_weight.values),