Fix CI

ae3dca77 · Yu Yang · f1913d46 · ae3dca77 · ae3dca77
隐藏空白更改
内联并排

Showing with 4 addition and 3 deletion

paddle/operators/scatter_op.cc paddle/operators/scatter_op.cc +2 -2

paddle/operators/softmax_with_cross_entropy_op.cc paddle/operators/softmax_with_cross_entropy_op.cc +2 -1

未找到文件。
--- a/paddle/operators/scatter_op.cc
+++ b/paddle/operators/scatter_op.cc
@@ -51,7 +51,7 @@ class ScatterOp : public framework::OperatorWithKernel {

  framework::DataType IndicateDataType(
      const framework::ExecutionContext& ctx) const override {
-    return framework::ToDataType(ctx.Input<Tensor>("X")->type());
+    return framework::ToDataType(ctx.Input<Tensor>("Ref")->type());
  }
 };

@@ -68,7 +68,7 @@ class ScatterGradOp : public framework::OperatorWithKernel {

  framework::DataType IndicateDataType(
      const framework::ExecutionContext& ctx) const override {
-    return framework::ToDataType(ctx.Input<Tensor>("X")->type());
+    return framework::ToDataType(ctx.Input<Tensor>("Ref")->type());
  }
 };


--- a/paddle/operators/softmax_with_cross_entropy_op.cc
+++ b/paddle/operators/softmax_with_cross_entropy_op.cc
@@ -158,7 +158,8 @@ class SoftmaxWithCrossEntropyOpGrad : public framework::OperatorWithKernel {

  framework::DataType IndicateDataType(
      const framework::ExecutionContext& ctx) const override {
-    return framework::ToDataType(ctx.Input<Tensor>("Logits")->type());
+    return framework::ToDataType(
+        ctx.Input<Tensor>(framework::GradVarName("Loss"))->type());
  }
 };