Merge pull request #16634 from junjun315/my-cool-stuff

fix load bug about context.place
6 years ago · 46e18cb700
parent d3b6291035
commit 46e18cb700
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;
  }
 };