diff --git a/paddle/fluid/inference/analysis/ir_passes/tensorrt_subgraph_pass.cc b/paddle/fluid/inference/analysis/ir_passes/tensorrt_subgraph_pass.cc index 4bd804dfca4d56cc7cb78984f12ae6d2233f77b8..c84bba33be14832895a181660051ad1c1d7f39aa 100644 --- a/paddle/fluid/inference/analysis/ir_passes/tensorrt_subgraph_pass.cc +++ b/paddle/fluid/inference/analysis/ir_passes/tensorrt_subgraph_pass.cc @@ -392,7 +392,7 @@ REGISTER_PASS_CAPABILITY(tensorrt_subgraph_pass) .EQ("shuffle_channel", 0) .EQ("swish", 0) .EQ("split", 0) - .EQ("instance_norm", 0) + .LE("instance_norm", 1) .EQ("gelu", 0) .EQ("layer_norm", 0) .EQ("scale", 0) diff --git a/paddle/fluid/operators/instance_norm_op.cc b/paddle/fluid/operators/instance_norm_op.cc index 1018adcd930a4a81fbb46afb35c035f6b4c8df9b..28643ac1c0d832dd9550c036f0a08383b256b5f6 100644 --- a/paddle/fluid/operators/instance_norm_op.cc +++ b/paddle/fluid/operators/instance_norm_op.cc @@ -17,6 +17,7 @@ limitations under the License. */ #include #include #include "paddle/fluid/framework/data_layout.h" +#include "paddle/fluid/framework/op_version_registry.h" #include "paddle/fluid/operators/math/math_function.h" namespace paddle { @@ -701,3 +702,20 @@ REGISTER_OP_CPU_KERNEL( float>, ops::InstanceNormDoubleGradKernel); + +REGISTER_OP_VERSION(instance_norm) + .AddCheckpoint( + R"ROC( + Change dispensable of attribute from False to True in instance_norm. + )ROC", + paddle::framework::compatible::OpVersionDesc() + .ModifyAttr( + "Bias", + "The arg 'dispensable' of Input 'Bias' is changed: from " + "'False' to 'True'.", + true) + .ModifyAttr( + "Scale", + "The arg 'dispensable' of Input 'Scale' is changed: from " + "'False' to 'True'.", + true));