From 44b226eda62ea4641f6eedd618e54a241e24ef28 Mon Sep 17 00:00:00 2001 From: heqiaozhi Date: Wed, 10 Apr 2019 16:13:24 +0800 Subject: [PATCH] ctx.device_context() to CPUPlace test=develop --- paddle/fluid/operators/cvm_op.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/operators/cvm_op.cc b/paddle/fluid/operators/cvm_op.cc index 662fb3b08e..b02cf10230 100644 --- 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("X")->type(), - ctx.device_context()); + platform::CPUPlace()); } }; -- GitLab