From ab4f1fa8e278b1f14ee2bd0bea4be5ad44ff4651 Mon Sep 17 00:00:00 2001 From: KerfuffleV2 Date: Sun, 7 May 2023 03:52:37 -0600 Subject: [PATCH] Output correct error message in ggml_new_tensor_impl when out of scratch space. --- ggml.c | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ggml.c b/ggml.c index b8a70599a..d3d73c874 100644 --- a/ggml.c +++ b/ggml.c @@ -4683,7 +4683,7 @@ struct ggml_tensor * ggml_new_tensor_impl( size_t size_needed = 0; if (!ggml_tensor_required_memory(ctx, type, n_dims, ne, data, &ctx_needed, &scratch_needed)) { - if (scratch_needed > 0 && ctx->scratch.size) { + if (scratch_needed > 0 && ctx->scratch.offs + scratch_needed > ctx->scratch.size) { GGML_PRINT("%s: not enough space in the scratch memory (needed %zu, available %zu)\n", __func__, ctx->scratch.offs + scratch_needed, ctx->scratch.size); } else {