Skip to content

Commit 7a7458f

Browse files
[OpenCL] Fix concat setArgs err (#6540)
1 parent 6aee599 commit 7a7458f

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

lite/kernels/opencl/concat_image_compute.cc

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -209,10 +209,10 @@ class ConcatComputeImage : public KernelLite<TARGET(kOpenCL),
209209
int arg_cnt = 0;
210210
for (auto& input : inputs) {
211211
auto* input_image_p = GET_DATA_GPU(input);
212-
kernel.setArg(arg_cnt++, *input_image_p);
212+
status = kernel.setArg(arg_cnt++, *input_image_p);
213213
CL_CHECK_FATAL(status);
214214
}
215-
kernel.setArg(arg_cnt++, *output_image_p);
215+
status = kernel.setArg(arg_cnt++, *output_image_p);
216216
CL_CHECK_FATAL(status);
217217

218218
for (auto& input : inputs) {
@@ -221,15 +221,15 @@ class ConcatComputeImage : public KernelLite<TARGET(kOpenCL),
221221
in_shape.s[4 - input->dims().size() + j] = input->dims()[j];
222222
}
223223
in_shape.s[1] = UP_DIV(in_shape.s[1], 4);
224-
kernel.setArg(arg_cnt++, in_shape);
224+
status = kernel.setArg(arg_cnt++, in_shape);
225225
CL_CHECK_FATAL(status);
226226
}
227227
cl_int4 out_shape = {1, 1, 1, 1};
228228
for (auto j = 0; j < output_tensor_dims.size(); ++j) {
229229
out_shape.s[4 - output_tensor_dims.size() + j] = output_tensor_dims[j];
230230
}
231231
out_shape.s[1] = UP_DIV(out_shape.s[1], 4);
232-
kernel.setArg(arg_cnt++, out_shape);
232+
status = kernel.setArg(arg_cnt++, out_shape);
233233
CL_CHECK_FATAL(status);
234234

235235
status = EnqueueNDRangeKernel(context,

0 commit comments

Comments
 (0)