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

6 years ago · d8a9b134e3
parent cc1a9f4238
commit d8a9b134e3
1 changed files with 2 additions and 1 deletions
--- 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