File tree Expand file tree Collapse file tree 2 files changed +3
-8
lines changed
paddle/fluid/pir/dialect/operator/interface Expand file tree Collapse file tree 2 files changed +3
-8
lines changed Original file line number Diff line number Diff line change @@ -104,13 +104,8 @@ common::DataLayout PreferLayoutImpl<Conv2dOp>(pir::Operation* op) {
104
104
}
105
105
106
106
template <>
107
- std::vector<pir::Value> RelevantInputsImpl<Conv2dOp>(pir::Operation* op) {
108
- // Note(lyk): We exhibit the layout transformation for filter of conv2d
109
- // due to issues with its infermeta and kernel not functioning
110
- // properly in NHWC layout. However, if the FLAGS_manually_trans_conv_filter
111
- // is enabled, the transfer_layout_pass can also operate correctly.
112
- auto concrete_op = op->dyn_cast <Conv2dOp>();
113
- return {concrete_op.input ()};
107
+ bool CanBeModifiedImpl<Conv2dOp>(pir::Operation* op) {
108
+ return false ;
114
109
}
115
110
116
111
template <>
Original file line number Diff line number Diff line change @@ -109,7 +109,7 @@ OVERLOAD_CAN_BE_MODIFIED(FusedConv2dAddActOp);
109
109
110
110
class Conv2dOp ;
111
111
OVERLOAD_PREFER_LAYOUT (Conv2dOp);
112
- OVERLOAD_RELEVANT_INPUTS (Conv2dOp);
112
+ OVERLOAD_CAN_BE_MODIFIED (Conv2dOp);
113
113
OVERLOAD_REWRITE_BY_LAYOUT (Conv2dOp);
114
114
115
115
class GroupNormOp ;
You can’t perform that action at this time.
0 commit comments