ggml : fix bug in ggml_concat

This commit is contained in:
Georgi Gerganov 2023-12-07 19:06:48 +02:00
parent 6f01e9ecf7
commit 3d154ad283
No known key found for this signature in database
GPG key ID: 449E073F9DC10735

2
ggml.c
View file

@ -8468,7 +8468,7 @@ static void ggml_compute_forward_concat_f32(
GGML_ASSERT(nb10 == sizeof(float));
for (int i3 = 0; i3 < ne3; i3++) {
for (int i2 = ith; i2 < ne2; i2++) {
for (int i2 = ith; i2 < ne2; i2 += nth) {
if (i2 < ne02) { // src0
for (int i1 = 0; i1 < ne1; i1++) {
for (int i0 = 0; i0 < ne0; i0++) {