Enable device automatically switching in mine_hard_examples_op. (#8706)

PaddlePaddle · Mar 5, 2018 · 457d012 · 457d012
1 parent bbf98a0
commit 457d012
Showing 1 changed file with 1 addition and 1 deletion.
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());
   }
 };