fix a skip_layernorm bug, test=develop (#26800)

ac63c7cd · zlsh80826 · GitHub · 994217ea · ac63c7cd
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/inference/tensorrt/plugin/skip_layernorm_op_plugin.h ...luid/inference/tensorrt/plugin/skip_layernorm_op_plugin.h +1 -1

未找到文件。
--- a/paddle/fluid/inference/tensorrt/plugin/skip_layernorm_op_plugin.h
+++ b/paddle/fluid/inference/tensorrt/plugin/skip_layernorm_op_plugin.h
@@ -54,7 +54,7 @@ class SkipLayerNormPluginDynamic : public DynamicPluginTensorRT {
    auto ptr = new SkipLayerNormPluginDynamic(
        bias_.data(), scale_.data(), bias_size_, scale_size_, eps_, ban_fp16_);
    ptr->bias_gpu_ = bias_gpu_;
-    ptr->scale_gpu_ = bias_gpu_;
+    ptr->scale_gpu_ = scale_gpu_;
    return ptr;
  }