@@ -916,11 +916,12 @@ static void RegisterOperatorKernel(
916
916
OperatorWithKernel::OpKernelFunc op_kernel_func;
917
917
if (kernel_func) {
918
918
VLOG (3 ) << " Register custom operator " << name << " with kernel func" ;
919
- op_kernel_func = [kernel_func, inputs, outputs, attrs, inplace_map](
920
- const framework::ExecutionContext& ctx) {
921
- VLOG (3 ) << " Custom Operator: run custom kernel func in lambda." ;
922
- RunKernelFunc (ctx, kernel_func, inputs, outputs, attrs, inplace_map);
923
- };
919
+ op_kernel_func =
920
+ [kernel_func, inputs, outputs, attrs, inplace_map]( // NOLINT
921
+ const framework::ExecutionContext& ctx) {
922
+ VLOG (3 ) << " Custom Operator: run custom kernel func in lambda." ;
923
+ RunKernelFunc (ctx, kernel_func, inputs, outputs, attrs, inplace_map);
924
+ };
924
925
} else {
925
926
VLOG (3 ) << " Register custom operator " << name
926
927
<< " with raw op kernel func" ;
@@ -1027,12 +1028,12 @@ void RegisterOperatorWithMetaInfo(const std::vector<OpMetaInfo>& op_meta_infos,
1027
1028
// InferShape
1028
1029
if (infer_shape_func == nullptr ) {
1029
1030
// use default InferShape
1030
- info.infer_shape_ =
1031
- [op_inputs, op_outputs, op_inplace_map]( InferShapeContext* ctx) {
1032
- RunDefaultInferShapeFunc (ctx, op_inputs, op_outputs, op_inplace_map);
1033
- };
1031
+ info.infer_shape_ = [op_inputs, op_outputs, op_inplace_map]( // NOLINT
1032
+ InferShapeContext* ctx) {
1033
+ RunDefaultInferShapeFunc (ctx, op_inputs, op_outputs, op_inplace_map);
1034
+ };
1034
1035
} else {
1035
- info.infer_shape_ = [op_inputs,
1036
+ info.infer_shape_ = [op_inputs, // NOLINT
1036
1037
op_outputs,
1037
1038
op_attrs,
1038
1039
op_inplace_map,
@@ -1051,12 +1052,12 @@ void RegisterOperatorWithMetaInfo(const std::vector<OpMetaInfo>& op_meta_infos,
1051
1052
// Infer Dtype
1052
1053
if (infer_dtype_func == nullptr ) {
1053
1054
// use default InferDtype
1054
- info.infer_var_type_ =
1055
- [op_inputs, op_outputs, op_inplace_map]( InferVarTypeContext* ctx) {
1056
- RunDefaultInferDtypeFunc (ctx, op_inputs, op_outputs, op_inplace_map);
1057
- };
1055
+ info.infer_var_type_ = [op_inputs, op_outputs, op_inplace_map]( // NOLINT
1056
+ InferVarTypeContext* ctx) {
1057
+ RunDefaultInferDtypeFunc (ctx, op_inputs, op_outputs, op_inplace_map);
1058
+ };
1058
1059
} else {
1059
- info.infer_var_type_ = [op_inputs,
1060
+ info.infer_var_type_ = [op_inputs, // NOLINT
1060
1061
op_outputs,
1061
1062
op_attrs,
1062
1063
op_inplace_map,
@@ -1115,7 +1116,10 @@ void RegisterOperatorWithMetaInfo(const std::vector<OpMetaInfo>& op_meta_infos,
1115
1116
1116
1117
// GradOpDescMaker
1117
1118
info.grad_op_maker_ =
1118
- [grad_op_name, grad_op_inputs, grad_op_outputs, is_double_grad](
1119
+ [grad_op_name, // NOLINT
1120
+ grad_op_inputs,
1121
+ grad_op_outputs,
1122
+ is_double_grad](
1119
1123
const OpDesc& fwd_op,
1120
1124
const std::unordered_set<std::string>& no_grad_set,
1121
1125
std::unordered_map<std::string, std::string>* grad_to_var,
@@ -1133,7 +1137,10 @@ void RegisterOperatorWithMetaInfo(const std::vector<OpMetaInfo>& op_meta_infos,
1133
1137
1134
1138
// GradOpBaseMaker
1135
1139
info.dygraph_grad_op_maker_ =
1136
- [grad_op_name, grad_op_inputs, grad_op_outputs, is_double_grad](
1140
+ [grad_op_name, // NOLINT
1141
+ grad_op_inputs,
1142
+ grad_op_outputs,
1143
+ is_double_grad](
1137
1144
const std::string& type,
1138
1145
const imperative::NameVarBaseMap& var_base_map_in,
1139
1146
const imperative::NameVarBaseMap& var_base_map_out,
@@ -1173,7 +1180,7 @@ void RegisterOperatorWithMetaInfo(const std::vector<OpMetaInfo>& op_meta_infos,
1173
1180
1174
1181
// Grad InferShape
1175
1182
if (grad_infer_shape_fn == nullptr ) {
1176
- grad_info.infer_shape_ = [grad_op_inputs,
1183
+ grad_info.infer_shape_ = [grad_op_inputs, // NOLINT
1177
1184
grad_op_outputs,
1178
1185
is_double_grad](InferShapeContext* ctx) {
1179
1186
// 1. if forward input exists, gradient's shape is same with forward
@@ -1211,7 +1218,7 @@ void RegisterOperatorWithMetaInfo(const std::vector<OpMetaInfo>& op_meta_infos,
1211
1218
}
1212
1219
};
1213
1220
} else {
1214
- grad_info.infer_shape_ = [grad_op_inputs,
1221
+ grad_info.infer_shape_ = [grad_op_inputs, // NOLINT
1215
1222
grad_op_outputs,
1216
1223
grad_op_attrs,
1217
1224
grad_op_inplace_map,
@@ -1230,7 +1237,7 @@ void RegisterOperatorWithMetaInfo(const std::vector<OpMetaInfo>& op_meta_infos,
1230
1237
// Grad InferDtype
1231
1238
if (grad_infer_dtype_fn != nullptr ) {
1232
1239
grad_info.infer_var_type_ =
1233
- [grad_op_inputs,
1240
+ [grad_op_inputs, // NOLINT
1234
1241
grad_op_outputs,
1235
1242
grad_op_attrs,
1236
1243
grad_op_inplace_map,
0 commit comments