Skip to content

Commit

Permalink
[InferSymbolicShape] Delete redundent value_id_to_shapeordata_ (#60554)
Browse files Browse the repository at this point in the history
  • Loading branch information
jiahy0825 committed Jan 4, 2024
1 parent c2dd202 commit 1d59851
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 5 deletions.
4 changes: 2 additions & 2 deletions paddle/pir/dialect/shape/utils/shape_utils.cc
Original file line number Diff line number Diff line change
Expand Up @@ -178,13 +178,13 @@ std::string GetValueId(Value* val) {
const symbol::ShapeOrDataDimExprs&
ShapeConstraintIRAnalysis::GetShapeOrDataForValue(Value* val) {
auto val_id = GetValueId(val);
return value_id_to_shapeordata[val_id];
return value_id_to_shapeordata_[val_id];
}

void ShapeConstraintIRAnalysis::SetShapeOrDataForValue(
Value* val, const symbol::ShapeOrDataDimExprs& shape_or_data) {
auto val_id = GetValueId(val);
value_id_to_shapeordata[val_id] = shape_or_data;
value_id_to_shapeordata_[val_id] = shape_or_data;
}

} // namespace pir
3 changes: 0 additions & 3 deletions paddle/pir/dialect/shape/utils/shape_utils.h
Original file line number Diff line number Diff line change
Expand Up @@ -105,9 +105,6 @@ class IR_API ShapeConstraintIRAnalysis : public ShapeAnalysis {
int64_t next_sym_idx_ = 0;
std::vector<symbol::DimExprConstraint> constraints_;

std::unordered_map<std::string, symbol::ShapeOrDataDimExprs>
value_id_to_shapeordata;

public:
explicit ShapeConstraintIRAnalysis(std::shared_ptr<pir::Program>&& program)
: ShapeConstraintIRAnalysis(program->module_op()) {
Expand Down

0 comments on commit 1d59851

Please sign in to comment.