Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
32 changes: 32 additions & 0 deletions paddle/fluid/pir/dialect/operator/ir/ops.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,8 @@
func : add
inplace : (x -> out)
backward : add_grad
data_transform :
support_trans_dtype : x, y
interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : add_n
Expand Down Expand Up @@ -373,6 +375,8 @@
kernel :
func : divide
inplace: (x -> out)
data_transform :
support_trans_dtype : x, y
backward : divide_grad

- op : dropout
Expand Down Expand Up @@ -406,6 +410,8 @@
spmd_rule: ElementwiseBinaryInferSpmd
kernel :
func : elementwise_pow
data_transform :
support_trans_dtype : x, y
backward : elementwise_pow_grad

- op : embedding
Expand Down Expand Up @@ -464,6 +470,8 @@
spmd_rule: ElementwiseBinaryInferSpmd
kernel :
func : equal
data_transform :
support_trans_dtype : x, y
inplace: (x -> out)

- op : exponential_
Expand Down Expand Up @@ -511,6 +519,8 @@
func : ElementwiseInferMeta
kernel :
func : floor_divide
data_transform :
support_trans_dtype : x, y
inplace: (x -> out)

- op : frobenius_norm
Expand Down Expand Up @@ -654,6 +664,8 @@
func : CompareInferMeta
kernel :
func : greater_equal
data_transform :
support_trans_dtype : x, y
inplace: (x -> out)

- op : greater_than
Expand All @@ -663,6 +675,8 @@
func : CompareInferMeta
kernel :
func : greater_than
data_transform :
support_trans_dtype : x, y
inplace: (x -> out)

- op : hardswish
Expand Down Expand Up @@ -702,6 +716,8 @@
func : CompareInferMeta
kernel :
func : less_equal
data_transform :
support_trans_dtype : x, y
inplace: (x -> out)

- op : less_than
Expand All @@ -711,6 +727,8 @@
func : CompareInferMeta
kernel :
func : less_than
data_transform :
support_trans_dtype : x, y
inplace: (x -> out)

- op : linspace
Expand Down Expand Up @@ -776,6 +794,8 @@
spmd_rule : MatmulInferSpmd
kernel :
func : matmul
data_transform :
support_trans_dtype : x, y
backward : matmul_grad

- op : matrix_rank
Expand Down Expand Up @@ -813,6 +833,8 @@
spmd_rule : ElementwiseBinaryInferSpmd
kernel :
func : maximum
data_transform :
support_trans_dtype : x, y
backward : maximum_grad

- op : mean
Expand Down Expand Up @@ -869,6 +891,8 @@
func : ElementwiseInferMeta
kernel :
func : minimum
data_transform :
support_trans_dtype : x, y
backward : minimum_grad

- op : mish
Expand All @@ -891,6 +915,8 @@
func : multiply {dense, dense -> dense},
multiply_sr {selected_rows, dense -> selected_rows}
inplace : (x -> out)
data_transform :
support_trans_dtype : x, y
backward : multiply_grad

- op : norm
Expand All @@ -910,6 +936,8 @@
spmd_rule : ElementwiseBinaryInferSpmd
kernel :
func : not_equal
data_transform :
support_trans_dtype : x, y
inplace: (x -> out)

- op : one_hot
Expand Down Expand Up @@ -1045,6 +1073,8 @@
func : ElementwiseInferMeta
kernel :
func : remainder
data_transform :
support_trans_dtype : x, y
inplace : (x -> out)

- op : repeat_interleave
Expand Down Expand Up @@ -1262,6 +1292,8 @@
kernel :
func : subtract
inplace : (x -> out)
data_transform :
support_trans_dtype : x, y
backward : subtract_grad
interfaces : paddle::dialect::InferSymbolicShapeInterface

Expand Down