fix load type, test=develop

6 years ago · 3773260edd
parent 67cddc2543
commit 3773260edd
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/load_op.cc
+++ b/paddle/fluid/operators/load_op.cc
@ -29,7 +29,7 @@ class LoadOp : public framework::OperatorWithKernel {
  framework::OpKernelType GetExpectedKernelType(
      const framework::ExecutionContext &ctx) const override {
    framework::OpKernelType kt = framework::OpKernelType(
-        framework::proto::VarType::FP32, platform::CPUPlace());
+        framework::proto::VarType::FP32, ctx.GetPlace());
    return kt;
  }
 };