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

junjun315 · junjun315 · commit 46e18cb7000c · 2019-04-03T20:16:20.000+08:00
fix load bug about context.place
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`	`};`