[Infer Symbolic Shape No.240] Add symbol_infer_interface for uniform_random_batch_size_like (#68980)

* Add symbol_infer_interface for uniform_random_batch_size_like

* Fix

* Fix

* Fix

* Fix

* revert irrelevant change

Author: ooooo-create

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

@@ -4052,11 +4052,15 @@ bool UniformInplace_OpInferSymbolicShape(
   return UniformInplaceOpInferSymbolicShape(op, infer_context);
 }
 
-// bool UniformRandomBatchSizeLikeOpInferSymbolicShape(
-//     pir::Operation *op, pir::InferSymbolicShapeContext *infer_context) {
-//   // pass
-//   return true;
-// }
+bool UniformRandomBatchSizeLikeOpInferSymbolicShape(
+    pir::Operation *op, pir::InferSymbolicShapeContext *infer_context) {
+  return BatchSizeLikeInferSymbolicShape(op, infer_context);
+}
+
+bool UniformRandomBatchSizeLikeSrOpInferSymbolicShape(
+    pir::Operation *op, pir::InferSymbolicShapeContext *infer_context) {
+  return UniformRandomBatchSizeLikeOpInferSymbolicShape(op, infer_context);
+}
 
 bool UniqueOpInferSymbolicShape(pir::Operation *op,
                                 pir::InferSymbolicShapeContext *infer_context) {
@@ -4224,12 +4228,8 @@ bool UnsqueezeOpInferSymbolicShape(
   int x_dims_size = x_sym_shape.size();
 
   std::vector<symbol::DimExpr> axis_sym;
-  if (axis_shape_or_data.data().has_value()) {
-    axis_sym = axis_shape_or_data.data().value();
-  } else {
-    axis_sym =
-        details::GetOrCreateExprVecFromData(axis_shape_or_data, infer_context);
-  }
+  axis_sym =
+      details::GetOrCreateExprVecFromData(axis_shape_or_data, infer_context);
   int axis_sym_size = axis_sym.size();
 
   // GetUnsqueezeShape

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

@@ -156,7 +156,8 @@ OP_DECLARE_INFER_SYMBOLIC_SHAPE(TransLayout)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Unbind)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(UniformInplace)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(UniformInplace_)
-// OP_DECLARE_INFER_SYMBOLIC_SHAPE(UniformRandomBatchSizeLike)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(UniformRandomBatchSizeLike)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(UniformRandomBatchSizeLikeSr)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Unique)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(UniqueConsecutive)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Unsqueeze)

paddle/phi/ops/yaml/ops.yaml

@@ -5190,7 +5190,7 @@
     data_type: dtype
   no_need_buffer: input
   traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait
-  # interfaces : paddle::dialect::InferSymbolicShapeInterface
+  interfaces : paddle::dialect::InferSymbolicShapeInterface
 
 - op : unique_consecutive
   args : (Tensor x, bool return_inverse = false, bool return_counts = false, int[] axis = {}, DataType dtype = DataType::FLOAT32)