Merge pull request #992 from dolphin8/metal

concat

Merge pull request #992 from dolphin8/metal
concat
b04a6bb5 · dolphin8 · GitHub · 3b4ba914 · 429309fb · b04a6bb5
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

metal/paddle-mobile/paddle-mobile/Operators/Kernels/metal/ConcatKernel.inc.metal ...dle-mobile/Operators/Kernels/metal/ConcatKernel.inc.metal +1 -1

未找到文件。
--- 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<P, access::read> in0 [[tex
       }
       k -= cp.vdim[j];
     }
-     if (k > cp.vdim[N-1]) {
+     if (j == N) {
       continue;
     }
     int ta = cp.odim[cp.axis];