Skip to content

Commit 40c1df6

Browse files
committed
add spmd rule for ont_hot and cumsum in yaml
1 parent 54866f6 commit 40c1df6

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

paddle/phi/ops/yaml/ops.yaml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1200,6 +1200,7 @@
12001200
output : Tensor(out)
12011201
infer_meta :
12021202
func : CumScalarAxisInferMeta
1203+
spmd_rule : CumSumInferSpmdDynamic
12031204
kernel :
12041205
func : cumsum
12051206
data_type : x
@@ -3491,6 +3492,7 @@
34913492
output : Tensor(out)
34923493
infer_meta :
34933494
func : OneHotInferMeta
3495+
spmd_rule : OneHotInferSpmdDynamic
34943496
kernel :
34953497
func : one_hot
34963498
traits : paddle::dialect::ForwardOnlyTrait

0 commit comments

Comments
 (0)