[CINN] Clear OpInferSymbolicShapeCache when lowering subgraph (#72248)

Author: huangjiyi

paddle/cinn/hlir/dialect/operator/transforms/lowering_pass/collect_sym_expr.cc

@@ -234,6 +234,7 @@ CreateGroupShapeOrDataExprs(
 
   local_shape_analysis.RegisterSymbolConstraintFromShapeAnalysis(
       global_shape_analysis);
+  local_shape_analysis.ClearOpInferSymbolicShapeCache();
   for (const auto& item : dim_expr_map) {
     local_shape_analysis.AddEqualCstr(item.first, item.second);
   }

paddle/pir/include/dialect/shape/utils/shape_analysis.h

@@ -136,6 +136,8 @@ class IR_API InferSymbolicShapeContext {
   std::optional<InferSymbolicShapeCacheValue> GetOpInferSymbolicShapeCache(
       const InferSymbolicShapeCacheKey& op_infer_cache_key) const;
 
+  void ClearOpInferSymbolicShapeCache();
+
   const symbol::ConstraintsManager& constraints_manager() const {
     return constraints_manager_;
   }
@@ -250,6 +252,10 @@ class IR_API ShapeConstraintIRAnalysis final
     return context_.constraints_manager();
   }
 
+  void ClearOpInferSymbolicShapeCache() {
+    context_.ClearOpInferSymbolicShapeCache();
+  }
+
   void SetInputDynamicDimSpec(
       const std::vector<InputDynamicDimSpec>& input_dynamic_dim_spec);
 

paddle/pir/src/dialect/shape/utils/shape_analysis.cc

@@ -480,6 +480,10 @@ InferSymbolicShapeContext::GetOpInferSymbolicShapeCache(
   return std::nullopt;
 }
 
+void InferSymbolicShapeContext::ClearOpInferSymbolicShapeCache() {
+  infer_symbolic_shape_cache_.clear();
+}
+
 bool InferSymbolicShapeContext::HasPredefinedDimExprForInputName(
     const std::string& input_name) const {
   return predefined_dimexpr_map_for_inputs_.count(input_name) != 0;