From ae3dca770c34d14ec81587eef9798c6a7070ec13 Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Wed, 27 Sep 2017 19:15:51 -0700
Subject: [PATCH] Fix CI

---
 paddle/operators/scatter_op.cc                    | 4 ++--
 paddle/operators/softmax_with_cross_entropy_op.cc | 3 ++-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/paddle/operators/scatter_op.cc b/paddle/operators/scatter_op.cc
index 619acfc8b62..cadd8841b6a 100644
--- 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());
   }
 };
 
diff --git a/paddle/operators/softmax_with_cross_entropy_op.cc b/paddle/operators/softmax_with_cross_entropy_op.cc
index de7c532421c..a76489871f3 100644
--- 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());
   }
 };
 
-- 
GitLab