From 83be1d9df3721f6ff876d330720fb700df1c83e1 Mon Sep 17 00:00:00 2001 From: dangqingqing Date: Wed, 7 Mar 2018 20:06:38 +0800 Subject: [PATCH] Fix detection_map_op for multi-device. --- paddle/fluid/operators/detection_map_op.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/operators/detection_map_op.cc b/paddle/fluid/operators/detection_map_op.cc index 9b8ca925373eb..73c84c2fe0155 100644 --- a/paddle/fluid/operators/detection_map_op.cc +++ b/paddle/fluid/operators/detection_map_op.cc @@ -71,7 +71,7 @@ class DetectionMAPOp : public framework::OperatorWithKernel { return framework::OpKernelType( framework::ToDataType( ctx.Input("DetectRes")->type()), - ctx.device_context()); + platform::CPUPlace()); } };