ctx.device_context() to CPUPlace

test=develop
6 years ago · 44b226eda6
parent fa6ea1e0e6
commit 44b226eda6
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/cvm_op.cc
+++ b/paddle/fluid/operators/cvm_op.cc
@ -94,7 +94,7 @@ class CVMGradientOp : public framework::OperatorWithKernel {
  framework::OpKernelType GetExpectedKernelType(
      const framework::ExecutionContext& ctx) const override {
    return framework::OpKernelType(ctx.Input<Tensor>("X")->type(),
-                                   ctx.device_context());
+                                   platform::CPUPlace());
  }
 };