Skip to content

Commit 292eca1

Browse files
【Infer Symbolic Shape BUAA No.91】Add viewshape op (#66737)
* Update unary_infer_sym.cc * Update unary_infer_sym.h * Update ops.yaml * Update same_operands_result.cc * Update unary_infer_sym.cc * Update unary_infer_sym.cc * Update same_operands_result.h * Update unary_infer_sym.h
1 parent 92a5f6b commit 292eca1

File tree

3 files changed

+3
-0
lines changed

3 files changed

+3
-0
lines changed

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -142,6 +142,7 @@ OP_SAME_OPERANDS_AND_RESULT(Trunc)
142142
OP_SAME_OPERANDS_AND_RESULT(Trunc_)
143143
OP_SAME_OPERANDS_AND_RESULT(Sigmoid)
144144
OP_SAME_OPERANDS_AND_RESULT(Sigmoid_)
145+
OP_SAME_OPERANDS_AND_RESULT(ViewShape)
145146

146147
bool ScaleOpInferSymbolicShape(pir::Operation *op,
147148
pir::InferSymbolicShapeContext *infer_context) {

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -135,6 +135,7 @@ OP_DECLARE_INFER_SYMBOLIC_SHAPE(Trunc)
135135
OP_DECLARE_INFER_SYMBOLIC_SHAPE(Trunc_)
136136
OP_DECLARE_INFER_SYMBOLIC_SHAPE(Sigmoid)
137137
OP_DECLARE_INFER_SYMBOLIC_SHAPE(Sigmoid_)
138+
OP_DECLARE_INFER_SYMBOLIC_SHAPE(ViewShape)
138139

139140
} // namespace paddle::dialect
140141

paddle/phi/ops/yaml/ops.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4824,6 +4824,7 @@
48244824
func : view_shape
48254825
backward : view_shape_grad
48264826
no_need_buffer : input
4827+
interfaces : paddle::dialect::InferSymbolicShapeInterface
48274828

48284829
- op : viterbi_decode
48294830
args : (Tensor potentials, Tensor transition_params, Tensor lengths, bool include_bos_eos_tag = true)

0 commit comments

Comments
 (0)