diff --git a/metal/paddle-mobile/paddle-mobile/Operators/Kernels/metal/ConcatKernel.inc.metal b/metal/paddle-mobile/paddle-mobile/Operators/Kernels/metal/ConcatKernel.inc.metal index b62daaa3aa1dd7861a3b00018bc38b409b05d8e0..c1af35de00ac2f26264fc1c6765f3e7405d3c0e5 100644 --- a/metal/paddle-mobile/paddle-mobile/Operators/Kernels/metal/ConcatKernel.inc.metal +++ b/metal/paddle-mobile/paddle-mobile/Operators/Kernels/metal/ConcatKernel.inc.metal @@ -65,7 +65,7 @@ kernel void FUNC(concat, R, N, VV, P)(texture2d_array
in0 [[tex } k -= cp.vdim[j]; } - if (k > cp.vdim[N-1]) { + if (j == N) { continue; } int ta = cp.odim[cp.axis];