Skip to content

Commit fa559d6

Browse files
committed
refactor(cinn): disable unused pool2d symbolic infer path
1 parent 0442435 commit fa559d6

3 files changed

Lines changed: 20 additions & 17 deletions

File tree

paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/cinn_op_infer_sym.cc

Lines changed: 15 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,9 @@
1515
#include "paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/cinn_op_infer_sym.h"
1616
#include "paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/infer_sym_slice_utils.h"
1717
#include "paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/infer_sym_utils.h"
18-
#include "paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/unary_infer_sym.cc"
18+
// clang-format off
19+
// #include "paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/unary_infer_sym.cc" // NOLINT
20+
// clang-format on
1921

2022
namespace cinn::dialect {
2123

@@ -222,17 +224,18 @@ bool SplitOpInferSymbolicShape(pir::Operation *op,
222224
return true;
223225
}
224226

225-
bool Pool2dOpInferSymbolicShape(pir::Operation *op,
226-
pir::InferSymbolicShapeContext *infer_context) {
227-
const auto &kernel_size_shape_or_data =
228-
infer_context->GetShapeOrDataForValue(op->operand_source(1));
229-
const auto &kernel_size =
230-
paddle::dialect::details::GetExprVecFromData(kernel_size_shape_or_data);
231-
infer_context->SetShapeOrDataForValue(
232-
op->result(0),
233-
Pool2dRawInferSymbolicShape(op, kernel_size, infer_context));
234-
return true;
235-
}
227+
// bool Pool2dOpInferSymbolicShape(pir::Operation *op,
228+
// pir::InferSymbolicShapeContext
229+
// *infer_context) {
230+
// const auto &kernel_size_shape_or_data =
231+
// infer_context->GetShapeOrDataForValue(op->operand_source(1));
232+
// const auto &kernel_size =
233+
// paddle::dialect::details::GetExprVecFromData(kernel_size_shape_or_data);
234+
// infer_context->SetShapeOrDataForValue(
235+
// op->result(0),
236+
// Pool2dRawInferSymbolicShape(op, kernel_size, infer_context));
237+
// return true;
238+
// }
236239

237240
bool ReduceInferSymbolicShape(pir::Operation *op,
238241
pir::InferSymbolicShapeContext *infer_context) {

paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/cinn_op_infer_sym.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
namespace cinn::dialect {
1919
OP_DECLARE_INFER_SYMBOLIC_SHAPE(Broadcast)
2020
OP_DECLARE_INFER_SYMBOLIC_SHAPE(Concat)
21-
OP_DECLARE_INFER_SYMBOLIC_SHAPE(Pool2d)
21+
// OP_DECLARE_INFER_SYMBOLIC_SHAPE(Pool2d)
2222
OP_DECLARE_INFER_SYMBOLIC_SHAPE(ReduceMax)
2323
OP_DECLARE_INFER_SYMBOLIC_SHAPE(ReduceMin)
2424
OP_DECLARE_INFER_SYMBOLIC_SHAPE(ReduceProd)

paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/unary_infer_sym.cc

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,8 @@ std::vector<symbol::DimExpr> GetRealPadding(
2121
const std::vector<int> &origin_paddings,
2222
const bool global_pooling,
2323
const bool adaptive,
24-
const std::string padding_algorithm,
25-
const std::vector<symbol::DimExpr> data_dims,
24+
const std::string &padding_algorithm,
25+
const std::vector<symbol::DimExpr> &data_dims,
2626
const std::vector<int> &strides,
2727
const std::vector<symbol::DimExpr> &kernel_size) {
2828
const auto &GetInitPadding = [&]() -> std::vector<symbol::DimExpr> {
@@ -51,7 +51,7 @@ std::vector<symbol::DimExpr> GetRealPadding(
5151

5252
std::vector<symbol::DimExpr> real_padding = GetInitPadding();
5353

54-
const auto &UpdataPadding = [&]() {
54+
const auto &UpdatePadding = [&]() {
5555
symbol::DimExpr one_dimexpr{1};
5656
symbol::DimExpr zero_dimexpr{0};
5757
// when padding_algorithm is "VALID" or "SAME"
@@ -81,7 +81,7 @@ std::vector<symbol::DimExpr> GetRealPadding(
8181
}
8282
};
8383

84-
UpdataPadding();
84+
UpdatePadding();
8585
return real_padding;
8686
}
8787

0 commit comments

Comments
 (0)