Skip to content

Commit ded34b2

Browse files
authored
Fix detection_map_op for multi-device. (#8845)
1 parent 7f00716 commit ded34b2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/fluid/operators/detection_map_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ class DetectionMAPOp : public framework::OperatorWithKernel {
7171
return framework::OpKernelType(
7272
framework::ToDataType(
7373
ctx.Input<framework::Tensor>("DetectRes")->type()),
74-
ctx.device_context());
74+
platform::CPUPlace());
7575
}
7676
};
7777

0 commit comments

Comments
 (0)