Fix

PaddlePaddle · May 19, 2024 · d03aa9f · d03aa9f
1 parent f1a09ac
commit d03aa9f
Show file tree

Hide file tree

Showing 18 changed files with 705 additions and 803 deletions.
diff --git a/paddle/fluid/operators/cross_entropy_op.cc b/paddle/fluid/operators/cross_entropy_op.cc
diff --git a/paddle/fluid/operators/cross_entropy_op.cu b/paddle/fluid/operators/cross_entropy_op.cu
diff --git a/paddle/fluid/operators/cross_entropy_op.h b/paddle/fluid/operators/cross_entropy_op.h
diff --git a/paddle/fluid/pir/dialect/op_generator/ops_api_gen.py b/paddle/fluid/pir/dialect/op_generator/ops_api_gen.py
@@ -197,6 +197,7 @@
     'c_reduce_prod',
     'c_reduce_prod_',
     'c_scatter',
+    "cross_entropy_grad2",
     'prune_gate_by_capacity',
     'push_sparse_v2',
     'push_sparse_v2_',

diff --git a/paddle/phi/api/yaml/backward.yaml b/paddle/phi/api/yaml/backward.yaml
@@ -2880,6 +2880,26 @@
     func : yolo_loss_grad
   optional : gt_score
 
+- backward_op: cross_entropy_grad
+  forward: cross_entropy (Tensor x, Tensor label, bool soft_label = false, int ignore_index = -100) -> Tensor (out)
+  args: (Tensor x, Tensor label, Tensor out_grad, bool soft_label = false, int ignore_index = -100)
+  output: Tensor (x_grad)
+  infer_meta:
+    func: CrossEntropyGradInferMeta
+  kernel:
+    func: cross_entropy_grad
+    data_type: out_grad
+
+- backward_op: cross_entropy_grad2
+  forward: cross_entropy2 (Tensor x, Tensor label, int ignore_index = -100) -> Tensor (out), Tensor (x_shape), Tensor (match_x)
+  args: (Tensor x_shape, Tensor label, Tensor match_x, Tensor out_grad, int ignore_index = -100)
+  output: Tensor (x_grad)
+  infer_meta:
+    func: CrossEntropyGrad2InferMeta
+  kernel:
+    func: cross_entropy_grad2
+    data_type: x_shape
+
 - backward_op: silu_double_grad
   forward: silu_grad (Tensor x, Tensor out, Tensor grad_out) -> Tensor(grad_x)
   args: (Tensor x, Tensor out, Tensor grad_out, Tensor grad_x_grad)