From d8a9b134e342d29a878bb8bc5ecfd9f5207850c4 Mon Sep 17 00:00:00 2001 From: Pei Yang Date: Wed, 8 Jan 2020 11:17:44 +0800 Subject: [PATCH] fix trt instance_norm serialize bug. test=develop (#22152) --- .../fluid/inference/tensorrt/plugin/instance_norm_op_plugin.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/inference/tensorrt/plugin/instance_norm_op_plugin.h b/paddle/fluid/inference/tensorrt/plugin/instance_norm_op_plugin.h index 3f5023f369..ec1d8e6517 100644 --- 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 -- GitLab