Skip to content

Commit a14b83c

Browse files
committed
update to latest mainline llama.cpp
add max_size param to ggml_metal_add_buffer - introduced in ggml-org/llama.cpp#1826
1 parent 3417a37 commit a14b83c

File tree

2 files changed

+7
-6
lines changed

2 files changed

+7
-6
lines changed

gpt4all-backend/replit.cpp

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -517,19 +517,20 @@ bool replit_model_load(const std::string & fname, std::istream &fin, replit_mode
517517
model.ctx_metal = ggml_metal_init();
518518
void* data_ptr = ggml_get_mem_buffer(model.ctx);
519519
size_t data_size = ggml_get_mem_size(model.ctx);
520+
const size_t max_size = ggml_get_max_tensor_size(model.ctx);
520521

521522
#define GGML_CHECK_BUF(result) if (!(result)) { \
522523
std::cerr << __func__ << ": failed to add buffer" << std::endl; \
523524
ggml_free(model.ctx); \
524525
return false; \
525526
}
526527

527-
GGML_CHECK_BUF(ggml_metal_add_buffer(model.ctx_metal, "data", data_ptr, data_size));
528+
GGML_CHECK_BUF(ggml_metal_add_buffer(model.ctx_metal, "data", data_ptr, data_size, max_size));
528529
GGML_CHECK_BUF(ggml_metal_add_buffer(model.ctx_metal, "kv", ggml_get_mem_buffer(model.kv_self.ctx),
529-
ggml_get_mem_size(model.kv_self.ctx)));
530-
GGML_CHECK_BUF(ggml_metal_add_buffer(model.ctx_metal, "eval", model.eval_buf, model.eval_buf_size));
531-
GGML_CHECK_BUF(ggml_metal_add_buffer(model.ctx_metal, "scr0", model.scr0_buf, model.scr0_buf_size));
532-
GGML_CHECK_BUF(ggml_metal_add_buffer(model.ctx_metal, "scr1", model.scr1_buf, model.scr1_buf_size));
530+
ggml_get_mem_size(model.kv_self.ctx), 0));
531+
GGML_CHECK_BUF(ggml_metal_add_buffer(model.ctx_metal, "eval", model.eval_buf, model.eval_buf_size, 0));
532+
GGML_CHECK_BUF(ggml_metal_add_buffer(model.ctx_metal, "scr0", model.scr0_buf, model.scr0_buf_size, 0));
533+
GGML_CHECK_BUF(ggml_metal_add_buffer(model.ctx_metal, "scr1", model.scr1_buf, model.scr1_buf_size, 0));
533534
#endif
534535

535536
return true;

0 commit comments

Comments
 (0)