diff --git a/paddle/fluid/framework/paddle2cinn/build_cinn_pass.h b/paddle/fluid/framework/paddle2cinn/build_cinn_pass.h index 4d7b784e32cedd3a1fa765bc891ce32e6c0a168e..b16c0a53d40cddfbf6b846e18fb03c48f968d44d 100644 --- a/paddle/fluid/framework/paddle2cinn/build_cinn_pass.h +++ b/paddle/fluid/framework/paddle2cinn/build_cinn_pass.h @@ -79,7 +79,8 @@ class OpTransInfo { DeParamCondT deny_param_cond_{{"batch_norm", {"ReserveSpace"}}, {"batch_norm_grad", {"ReserveSpace"}}}; - std::unordered_set default_deny_ops_{"feed", "fetch"}; + std::unordered_set default_deny_ops_{ + "feed", "fetch", "conv2d", "conv2d_grad"}; }; // A pass named BuildCinnPass, the function of this pass is: