ggml : fix bug introduced in #1237

This commit is contained in:
Georgi Gerganov 2023-07-05 20:44:11 +03:00
parent 1b6efeab82
commit ec326d350c
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735

2
ggml.c
View File

@ -16202,7 +16202,7 @@ void ggml_graph_compute(struct ggml_context * ctx, struct ggml_cgraph * cgraph)
if (node->src1->type != vec_dot_type) { if (node->src1->type != vec_dot_type) {
cur = GGML_TYPE_SIZE[vec_dot_type]*ggml_nelements(node->src1)/GGML_BLCK_SIZE[vec_dot_type]; cur = GGML_TYPE_SIZE[vec_dot_type]*ggml_nelements(node->src1)/GGML_BLCK_SIZE[vec_dot_type];
} else { } else {
GGML_ASSERT(false); cur = 0;
} }
work_size = MAX(work_size, cur); work_size = MAX(work_size, cur);