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

junjun315 · web-flow · commit af07056c06dc · 2019-04-04T12:26:26.000+08:00
cherry-pick:load_op.cc hot fix
diff --git 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;
   }
 };

Original file line number	Diff line number	Diff line change
`@@ -29,7 +29,7 @@ class LoadOp : public framework::OperatorWithKernel {`
`29`	`29`	`framework::OpKernelType GetExpectedKernelType(`
`30`	`30`	`const framework::ExecutionContext &ctx) const override {`
`31`	`31`	`framework::OpKernelType kt = framework::OpKernelType(`
`32`		`- framework::proto::VarType::FP32, platform::CPUPlace());`
	`32`	`+ framework::proto::VarType::FP32, ctx.GetPlace());`
`33`	`33`	`return kt;`
`34`	`34`	`}`
`35`	`35`	`};`