Enable device automatically switching in mine_hard_examples_op. (#8706)

qingqing01 · web-flow · commit 457d012a8e04 · 2018-03-05T13:48:40.000+08:00
diff --git a/paddle/fluid/operators/mine_hard_examples_op.cc b/paddle/fluid/operators/mine_hard_examples_op.cc
@@ -247,7 +247,7 @@ class MineHardExamplesOp : public framework::OperatorWithKernel {
       const framework::ExecutionContext& ctx) const override {
     return framework::OpKernelType(
         framework::ToDataType(ctx.Input<framework::Tensor>("ClsLoss")->type()),
-        ctx.device_context());
+        platform::CPUPlace());
   }
 };
 

Original file line number	Diff line number	Diff line change
`@@ -247,7 +247,7 @@ class MineHardExamplesOp : public framework::OperatorWithKernel {`
`247`	`247`	`const framework::ExecutionContext& ctx) const override {`
`248`	`248`	`return framework::OpKernelType(`
`249`	`249`	`framework::ToDataType(ctx.Input<framework::Tensor>("ClsLoss")->type()),`
`250`		`- ctx.device_context());`
	`250`	`+ platform::CPUPlace());`
`251`	`251`	`}`
`252`	`252`	`};`
`253`	`253`