Skip to content

Commit 90cc59d

Browse files
ikawrakowKawrakow
andauthored
examples : fix examples/metal (#1920)
Co-authored-by: Iwan Kawrakow <[email protected]>
1 parent ce2c7d7 commit 90cc59d

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

examples/metal/metal.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,10 @@ int main(int argc, char ** argv) {
4040
// this allocates all Metal resources and memory buffers
4141
auto * ctx_metal = ggml_metal_init();
4242

43-
ggml_metal_add_buffer(ctx_metal, "data", ggml_get_mem_buffer(ctx_data), ggml_get_mem_size(ctx_data));
44-
ggml_metal_add_buffer(ctx_metal, "eval", ggml_get_mem_buffer(ctx_eval), ggml_get_mem_size(ctx_eval));
43+
const size_t max_size_data = ggml_get_max_tensor_size(ctx_data);
44+
const size_t max_size_eval = ggml_get_max_tensor_size(ctx_eval);
45+
ggml_metal_add_buffer(ctx_metal, "data", ggml_get_mem_buffer(ctx_data), ggml_get_mem_size(ctx_data), max_size_data);
46+
ggml_metal_add_buffer(ctx_metal, "eval", ggml_get_mem_buffer(ctx_eval), ggml_get_mem_size(ctx_eval), max_size_eval);
4547

4648
// main
4749
{

0 commit comments

Comments
 (0)