metal : fix loop bound in ggml_mem_ranges (#16412)

This commit is contained in:
Georgi Gerganov
2025-10-03 19:18:56 +03:00
committed by GitHub
parent 946f71ed9a
commit 606a73f531

View File

@@ -112,7 +112,7 @@ static bool ggml_mem_ranges_add_dst(ggml_mem_ranges_t mrs, const ggml_tensor * t
} }
bool ggml_mem_ranges_add(ggml_mem_ranges_t mrs, const ggml_tensor * tensor) { bool ggml_mem_ranges_add(ggml_mem_ranges_t mrs, const ggml_tensor * tensor) {
for (int i = 0; i < GGML_MAX_DIMS; i++) { for (int i = 0; i < GGML_MAX_SRC; i++) {
if (tensor->src[i]) { if (tensor->src[i]) {
ggml_mem_ranges_add_src(mrs, tensor->src[i]); ggml_mem_ranges_add_src(mrs, tensor->src[i]);
} }
@@ -173,7 +173,7 @@ static bool ggml_mem_ranges_check_dst(ggml_mem_ranges_t mrs, const ggml_tensor *
} }
bool ggml_mem_ranges_check(ggml_mem_ranges_t mrs, const ggml_tensor * tensor) { bool ggml_mem_ranges_check(ggml_mem_ranges_t mrs, const ggml_tensor * tensor) {
for (int i = 0; i < GGML_MAX_DIMS; i++) { for (int i = 0; i < GGML_MAX_SRC; i++) {
if (tensor->src[i]) { if (tensor->src[i]) {
if (!ggml_mem_ranges_check_src(mrs, tensor->src[i])) { if (!ggml_mem_ranges_check_src(mrs, tensor->src[i])) {
return false; return false;