Skip to content

Commit 9099ea2

Browse files
committed
Add spmd rule for five ops(bitwise_or,atan2,fmax,fmin,reciprocal).
1 parent 746356c commit 9099ea2

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

paddle/phi/ops/yaml/backward.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2601,6 +2601,7 @@
26012601
infer_meta :
26022602
func : UnchangedInferMeta
26032603
param : [out]
2604+
spmd_rule : ElementwiseUnaryGradInferSpmd
26042605
kernel :
26052606
func : reciprocal_grad
26062607
inplace : (out_grad -> x_grad)

paddle/phi/ops/yaml/ops.yaml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -450,6 +450,7 @@
450450
output : Tensor(out)
451451
infer_meta :
452452
func : Atan2InferMeta
453+
spmd_rule : ElementwiseBinaryInferSpmd
453454
kernel :
454455
func : atan2
455456
backward : atan2_grad
@@ -689,6 +690,7 @@
689690
output : Tensor(out)
690691
infer_meta :
691692
func : ElementwiseInferMeta
693+
spmd_rule : ElementwiseBinaryInferSpmd
692694
kernel :
693695
func : bitwise_or
694696
backend : x
@@ -2074,6 +2076,7 @@
20742076
infer_meta :
20752077
param: [x, y]
20762078
func : ElementwiseInferMeta
2079+
spmd_rule : ElementwiseBinaryInferSpmd
20772080
kernel :
20782081
func : fmax
20792082
backward : fmax_grad
@@ -2086,6 +2089,7 @@
20862089
infer_meta :
20872090
func : ElementwiseInferMeta
20882091
param: [x, y]
2092+
spmd_rule : ElementwiseBinaryInferSpmd
20892093
kernel :
20902094
func : fmin
20912095
backward : fmin_grad
@@ -4092,6 +4096,7 @@
40924096
output : Tensor(out)
40934097
infer_meta :
40944098
func : UnchangedInferMeta
4099+
spmd_rule : ElementwiseUnaryInferSpmd
40954100
kernel :
40964101
func : reciprocal
40974102
inplace : (x -> out)

0 commit comments

Comments
 (0)