Skip to content

Commit 07e5a0f

Browse files
authored
[Fix](mluOpMsDeformAttnBackward): change global variable to local variable (Cambricon#1261)
1 parent 9055d9c commit 07e5a0f

1 file changed

Lines changed: 10 additions & 11 deletions

File tree

kernels/ms_deform_attn/ms_deform_attn_backward/ms_deform_attn_backward.cpp

Lines changed: 10 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -35,8 +35,6 @@
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

Comments
 (0)