fix auc op (#14385)

test=develop
6 years ago · 9f68e9a7fe
parent 38f499df7d
commit 9f68e9a7fe
2 changed files with 3 additions and 3 deletions
--- a/paddle/fluid/operators/auc_op.cc
+++ b/paddle/fluid/operators/auc_op.cc
@ -53,7 +53,7 @@ class AucOp : public framework::OperatorWithKernel {
      const framework::ExecutionContext &ctx) const override {
    return framework::OpKernelType(
        framework::ToDataType(ctx.Input<Tensor>("Predict")->type()),
-        ctx.device_context());
+        platform::CPUPlace());
  }
 };

--- a/paddle/fluid/operators/nce_op.cc
+++ b/paddle/fluid/operators/nce_op.cc
@ -69,7 +69,7 @@ class NCEOp : public framework::OperatorWithKernel {
      const framework::ExecutionContext& ctx) const override {
    return framework::OpKernelType(
        framework::ToDataType(ctx.Input<Tensor>("Input")->type()),
-        ctx.GetPlace());
+        platform::CPUPlace());
  }
 };

@ -174,7 +174,7 @@ class NCEOpGrad : public framework::OperatorWithKernel {
      const framework::ExecutionContext& ctx) const override {
    return framework::OpKernelType(
        framework::ToDataType(ctx.Input<Tensor>("Input")->type()),
-        ctx.GetPlace());
+        platform::CPUPlace());
  }
 };