3535#include " kernels/debug.h"
3636#include " kernels/kernel.h"
3737
38- char API [] = " [mluOpMsDeformAttnBackward]" ;
39-
4038#define MIN (a, b ) (((a) < (b)) ? (a) : (b))
4139
4240/* !
@@ -107,6 +105,7 @@ static mluOpStatus_t msDeformAttnBackwardParamCheck(
107105 const mluOpTensorDescriptor_t grad_attn_weight_desc, void *grad_attn_weight,
108106 bool *calc_grad_loc_weight_flag, bool *calc_grad_value_flag,
109107 bool *calc_grad_value_loc_weight_flag) {
108+ const char API [] = " [mluOpMsDeformAttnBackward]" ;
110109 // check desc
111110 PARAM_CHECK (API , handle != NULL );
112111 PARAM_CHECK (API , value_desc != NULL );
@@ -131,23 +130,23 @@ static mluOpStatus_t msDeformAttnBackwardParamCheck(
131130 PARAM_CHECK (API , grad_attn_weight_desc->getDim () == 5 );
132131
133132 // check stride
134- STRIDE_TENSOR_CHECK (" [mluOpMsDeformAttnBackward]: " , value_desc,
133+ STRIDE_TENSOR_CHECK (API , value_desc,
135134 " value_desc must be contiguous" );
136- STRIDE_TENSOR_CHECK (" [mluOpMsDeformAttnBackward]: " , spatial_shapes_desc,
135+ STRIDE_TENSOR_CHECK (API , spatial_shapes_desc,
137136 " spatial_shapes_desc must be contiguous" );
138- STRIDE_TENSOR_CHECK (" [mluOpMsDeformAttnBackward]: " , level_start_index_desc,
137+ STRIDE_TENSOR_CHECK (API , level_start_index_desc,
139138 " level_start_index_desc must be contiguous" );
140- STRIDE_TENSOR_CHECK (" [mluOpMsDeformAttnBackward]: " , sampling_loc_desc,
139+ STRIDE_TENSOR_CHECK (API , sampling_loc_desc,
141140 " sampling_loc_desc must be contiguous" );
142- STRIDE_TENSOR_CHECK (" [mluOpMsDeformAttnBackward]: " , attn_weight_desc,
141+ STRIDE_TENSOR_CHECK (API , attn_weight_desc,
143142 " attn_weight_desc must be contiguous" );
144- STRIDE_TENSOR_CHECK (" [mluOpMsDeformAttnBackward]: " , grad_output_desc,
143+ STRIDE_TENSOR_CHECK (API , grad_output_desc,
145144 " grad_output_desc must be contiguous" );
146- STRIDE_TENSOR_CHECK (" [mluOpMsDeformAttnBackward]: " , grad_value_desc,
145+ STRIDE_TENSOR_CHECK (API , grad_value_desc,
147146 " grad_value_desc must be contiguous" );
148- STRIDE_TENSOR_CHECK (" [mluOpMsDeformAttnBackward]: " , grad_sampling_loc_desc,
147+ STRIDE_TENSOR_CHECK (API , grad_sampling_loc_desc,
149148 " grad_sampling_loc_desc must be contiguous" );
150- STRIDE_TENSOR_CHECK (" [mluOpMsDeformAttnBackward]: " , grad_attn_weight_desc,
149+ STRIDE_TENSOR_CHECK (API , grad_attn_weight_desc,
151150 " grad_attn_weight_desc must be contiguous" );
152151
153152 // check datatype
0 commit comments