nikola/llama_cpp
1
0
Fork 0
Code Issues Pull Requests Actions 20 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Johannes Gäßler
2026-03-04 12:04:31 +01:00
committed by GitHub
parent 66199c9f03
commit 7f5ee54968
1 changed files with 7 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ggml/src/ggml.c
+1 -3
View File
@@ -1410,9 +1410,8 @@ static bool ggml_is_contiguous_n(const struct ggml_tensor * tensor, int n) {
} }
next_nb *= tensor->ne[0]/ggml_blck_size(tensor->type); next_nb *= tensor->ne[0]/ggml_blck_size(tensor->type);
for (int i = 1; i < GGML_MAX_DIMS; i++) { for (int i = 1; i < GGML_MAX_DIMS; i++) {
if (tensor->ne[i] != 1) {
if (i > n) { if (i > n) {
if (tensor->nb[i] != next_nb) { if (tensor->ne[i] != 1 && tensor->nb[i] != next_nb) {
return false; return false;
} }
next_nb *= tensor->ne[i]; next_nb *= tensor->ne[i];
@@ -1421,7 +1420,6 @@ static bool ggml_is_contiguous_n(const struct ggml_tensor * tensor, int n) {
next_nb = tensor->ne[i]*tensor->nb[i]; next_nb = tensor->ne[i]*tensor->nb[i];
} }
} }
}
return true; return true;
} }