Skip to content

Commit e2e1cb4

Browse files
committed
add spmd_rule for atan2_grad, fmax_grad, fmin_grad.
1 parent 9099ea2 commit e2e1cb4

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

paddle/phi/ops/yaml/backward.yaml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -215,6 +215,7 @@
215215
infer_meta :
216216
func : GeneralBinaryGradInferMeta
217217
param : [x, y]
218+
spmd_rule : ElementwiseBinaryGradInferSpmd
218219
kernel :
219220
func : atan2_grad
220221

@@ -1229,6 +1230,7 @@
12291230
infer_meta :
12301231
func : GeneralBinaryGradInferMeta
12311232
param: [x, y]
1233+
spmd_rule : ElementwiseBinaryGradInferSpmd
12321234
kernel :
12331235
func : fmax_grad
12341236
data_type : out_grad
@@ -1240,6 +1242,7 @@
12401242
infer_meta :
12411243
func : GeneralBinaryGradInferMeta
12421244
param: [x, y]
1245+
spmd_rule : ElementwiseBinaryGradInferSpmd
12431246
kernel :
12441247
func : fmin_grad
12451248
data_type : out_grad

0 commit comments

Comments
 (0)