From 3f2769bf261b6b9b5f36f9a180546d3a3a4ee2c1 Mon Sep 17 00:00:00 2001 From: ct-clmsn Date: Thu, 21 Dec 2023 18:25:22 -0500 Subject: [PATCH] added correct use of shmem_free --- ggml-oshmem.c | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/ggml-oshmem.c b/ggml-oshmem.c index 0ff2885ae..86e759693 100644 --- a/ggml-oshmem.c +++ b/ggml-oshmem.c @@ -32,7 +32,8 @@ void ggml_openshmem_backend_free(void) { } struct ggml_openshmem_context * ggml_openshmem_init(void) { - struct ggml_openshmem_context * ctx = calloc(1, sizeof(struct ggml_openshmem_context)); + struct ggml_openshmem_context * ctx = + (struct ggml_openshmem_context *)calloc(1, sizeof(struct ggml_openshmem_context)); ctx->pe = shmem_my_pe(); ctx->n_pes = shmem_n_pes(); @@ -62,6 +63,8 @@ struct ggml_openshmem_context * ggml_openshmem_init(void) { } void ggml_openshmem_free(struct ggml_openshmem_context * ctx) { + shmem_free(ctx->symmetric_comm_structure); + shmem_free(ctx->recv_signal); free(ctx); }