diff --git a/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/element_wise_binary.cc b/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/element_wise_binary.cc index 7c5c5b62fa9265..4d70ebbe69e8fb 100644 --- a/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/element_wise_binary.cc +++ b/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/element_wise_binary.cc @@ -137,6 +137,10 @@ bool SubtractOpInferSymbolicShape( OP_ELEMENT_WISE_BINARY(Add_) OP_ELEMENT_WISE_BINARY(BitwiseAnd) OP_ELEMENT_WISE_BINARY(BitwiseAnd_) +OP_ELEMENT_WISE_BINARY(BitwiseLeftShift) +OP_ELEMENT_WISE_BINARY(BitwiseLeftShift_) +OP_ELEMENT_WISE_BINARY(BitwiseOr) +OP_ELEMENT_WISE_BINARY(BitwiseOr_) OP_ELEMENT_WISE_BINARY(BitwiseXor) OP_ELEMENT_WISE_BINARY(BitwiseXor_) OP_ELEMENT_WISE_BINARY(Complex) diff --git a/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/element_wise_binary.h b/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/element_wise_binary.h index 0f8255f2132736..793cb8b55f9d0c 100644 --- a/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/element_wise_binary.h +++ b/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/element_wise_binary.h @@ -21,6 +21,10 @@ OP_DECLARE_INFER_SYMBOLIC_SHAPE(Add) OP_DECLARE_INFER_SYMBOLIC_SHAPE(Add_) OP_DECLARE_INFER_SYMBOLIC_SHAPE(BitwiseAnd) OP_DECLARE_INFER_SYMBOLIC_SHAPE(BitwiseAnd_) +OP_DECLARE_INFER_SYMBOLIC_SHAPE(BitwiseLeftShift) +OP_DECLARE_INFER_SYMBOLIC_SHAPE(BitwiseLeftShift_) +OP_DECLARE_INFER_SYMBOLIC_SHAPE(BitwiseOr) +OP_DECLARE_INFER_SYMBOLIC_SHAPE(BitwiseOr_) OP_DECLARE_INFER_SYMBOLIC_SHAPE(BitwiseXor) OP_DECLARE_INFER_SYMBOLIC_SHAPE(BitwiseXor_) OP_DECLARE_INFER_SYMBOLIC_SHAPE(Complex) diff --git a/paddle/phi/ops/yaml/ops.yaml b/paddle/phi/ops/yaml/ops.yaml index 683800a5cdcfca..bf8f9b7ccea834 100755 --- a/paddle/phi/ops/yaml/ops.yaml +++ b/paddle/phi/ops/yaml/ops.yaml @@ -588,6 +588,7 @@ backend : x inplace: (x -> out) traits : paddle::dialect::ForwardOnlyTrait + interfaces : paddle::dialect::InferSymbolicShapeInterface - op : bitwise_not args : (Tensor x) @@ -612,6 +613,7 @@ backend : x inplace: (x -> out) traits : paddle::dialect::ForwardOnlyTrait + interfaces : paddle::dialect::InferSymbolicShapeInterface - op : bitwise_right_shift args : (Tensor x, Tensor y, bool is_arithmetic = true)