Merge pull request #16671 from junjun315/cherry-pick-fix-load

cherry-pick:load_op.cc hot fix
6 years ago · af07056c06
parent ab6600e18c 46e18cb700
commit af07056c06
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;
  }
 };