From 6fbb723ec44da33881774cd728a113842a15843b Mon Sep 17 00:00:00 2001 From: xuxinyi Date: Tue, 11 Feb 2025 12:32:34 +0000 Subject: [PATCH] add_spmd --- paddle/phi/ops/yaml/backward.yaml | 1 + paddle/phi/ops/yaml/ops.yaml | 1 + 2 files changed, 2 insertions(+) diff --git a/paddle/phi/ops/yaml/backward.yaml b/paddle/phi/ops/yaml/backward.yaml index 2f15df6ecbbcbf..065f7426fc937d 100644 --- a/paddle/phi/ops/yaml/backward.yaml +++ b/paddle/phi/ops/yaml/backward.yaml @@ -2936,6 +2936,7 @@ infer_meta : func : UnchangedInferMeta param : [out] + spmd_rule : ElementwiseUnaryGradInferSpmd kernel : func : sigmoid_grad backward : sigmoid_double_grad diff --git a/paddle/phi/ops/yaml/ops.yaml b/paddle/phi/ops/yaml/ops.yaml index d77daee9429cd1..3b347576afbc62 100755 --- a/paddle/phi/ops/yaml/ops.yaml +++ b/paddle/phi/ops/yaml/ops.yaml @@ -4501,6 +4501,7 @@ output : Tensor infer_meta : func : UnchangedInferMeta + spmd_rule : ElementwiseUnaryInferSpmd kernel : func : sigmoid inplace : (x -> out)