ggml: fix ggml_is_contiguous_n for ne == 1 (#20092)

JohannesGaessler · web-flow · commit 7f5ee549683d · 2026-03-04T12:04:31.000+01:00
diff --git a/ggml/src/ggml.c b/ggml/src/ggml.c
@@ -1410,16 +1410,14 @@ static bool ggml_is_contiguous_n(const struct ggml_tensor * tensor, int n) {
     }
     next_nb *= tensor->ne[0]/ggml_blck_size(tensor->type);
     for (int i = 1; i < GGML_MAX_DIMS; i++) {
-        if (tensor->ne[i] != 1) {
-            if (i > n) {
-                if (tensor->nb[i] != next_nb) {
-                    return false;
-                }
-                next_nb *= tensor->ne[i];
-            } else {
-                // this dimension does not need to be contiguous
-                next_nb = tensor->ne[i]*tensor->nb[i];
+        if (i > n) {
+            if (tensor->ne[i] != 1 && tensor->nb[i] != next_nb) {
+                return false;
             }
+            next_nb *= tensor->ne[i];
+        } else {
+            // this dimension does not need to be contiguous
+            next_nb = tensor->ne[i]*tensor->nb[i];
         }
     }
     return true;

Original file line number	Diff line number	Diff line change
`@@ -1410,16 +1410,14 @@ static bool ggml_is_contiguous_n(const struct ggml_tensor * tensor, int n) {`
`1410`	`1410`	`}`
`1411`	`1411`	`next_nb *= tensor->ne[0]/ggml_blck_size(tensor->type);`
`1412`	`1412`	`for (int i = 1; i < GGML_MAX_DIMS; i++) {`
`1413`		`- if (tensor->ne[i] != 1) {`
`1414`		`- if (i > n) {`
`1415`		`- if (tensor->nb[i] != next_nb) {`
`1416`		`- return false;`
`1417`		`- }`
`1418`		`- next_nb *= tensor->ne[i];`
`1419`		`- } else {`
`1420`		`- // this dimension does not need to be contiguous`
`1421`		`- next_nb = tensor->ne[i]*tensor->nb[i];`
	`1413`	`+ if (i > n) {`
	`1414`	`+ if (tensor->ne[i] != 1 && tensor->nb[i] != next_nb) {`
	`1415`	`+ return false;`
`1422`	`1416`	`}`
	`1417`	`+ next_nb *= tensor->ne[i];`
	`1418`	`+ } else {`
	`1419`	`+ // this dimension does not need to be contiguous`
	`1420`	`+ next_nb = tensor->ne[i]*tensor->nb[i];`
`1423`	`1421`	`}`
`1424`	`1422`	`}`
`1425`	`1423`	`return true;`