vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

added correct use of shmem_free

Browse source
This commit is contained in:
ct-clmsn 2023-12-21 18:25:22 -05:00
parent 46bcbf3805
commit 3f2769bf26
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
ggml-oshmem.c
View file

@ -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);
}