fix trt instance_norm serialize bug. test=develop (#22152)

d8a9b134 · Pei Yang · GitHub · cc1a9f42 · d8a9b134
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

paddle/fluid/inference/tensorrt/plugin/instance_norm_op_plugin.h ...fluid/inference/tensorrt/plugin/instance_norm_op_plugin.h +2 -1

未找到文件。
--- a/paddle/fluid/inference/tensorrt/plugin/instance_norm_op_plugin.h
+++ b/paddle/fluid/inference/tensorrt/plugin/instance_norm_op_plugin.h
@@ -41,7 +41,8 @@ class InstanceNormPlugin : public PluginTensorRT {
 protected:
  size_t getSerializationSize() override {
    return getBaseSerializationSize() + SerializedSize(eps_) +
-           SerializedSize(scale_) + SerializedSize(bias_);
+           SerializedSize(scale_) + SerializedSize(bias_) +
+           SerializedSize(getPluginType());
  }

  // TRT will call this func when we need to serialize the configuration of