fix bug when run on GPU, test=develop

6 years ago · b10d84bc5a
parent 5fb2856584
commit b10d84bc5a
1 changed files with 8 additions and 1 deletions
--- a/paddle/fluid/operators/detection/box_clip_op.cc
+++ b/paddle/fluid/operators/detection/box_clip_op.cc
@ -41,6 +41,13 @@ class BoxClipOp : public framework::OperatorWithKernel {
    ctx->ShareDim("InputBox", /*->*/ "OutputBox");
    ctx->ShareLoD("InputBox", /*->*/ "OutputBox");
  }
+
+ protected:
+  framework::OpKernelType GetExpectedKernelType(
+      const framework::ExecutionContext& ctx) const override {
+    auto data_type = framework::GetDataTypeOfVar(ctx.InputVar("RpnRois"));
+    return framework::OpKernelType(data_type, platform::CPUPlace());
+  }
 };

 class BoxClipOpMaker : public framework::OpProtoAndCheckerMaker {