normalized custom operator impl (#32666)

PaddlePaddle · Apr 29, 2021 · 7a73692 · 7a73692
1 parent 10c493a
commit 7a73692
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddle/fluid/framework/custom_operator.cc b/paddle/fluid/framework/custom_operator.cc
@@ -246,7 +246,7 @@ class CustomOperator : public OperatorWithKernel {
    * it can only be determined at runtime.
    */
   framework::OpKernelType GetExpectedKernelType(
-      const framework::ExecutionContext& ctx) const {
+      const framework::ExecutionContext& ctx) const override {
     return framework::OpKernelType(proto::VarType::RAW, ctx.GetPlace());
   }
 
@@ -257,7 +257,7 @@ class CustomOperator : public OperatorWithKernel {
    */
   framework::OpKernelType GetKernelTypeForVar(
       const std::string& var_name, const Tensor& tensor,
-      const OpKernelType& expected_kernel_type) {
+      const OpKernelType& expected_kernel_type) const override {
     return OpKernelType(expected_kernel_type.data_type_,
                         expected_kernel_type.place_, tensor.layout());
   }