From f1ffb43e0b1f73057b38ecdf74524c60f4d8017b Mon Sep 17 00:00:00 2001 From: PatricZhao Date: Mon, 25 Oct 2021 14:15:11 +0800 Subject: [PATCH] Add free --- code/sync.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/code/sync.cpp b/code/sync.cpp index 0f0c97b..e4137b9 100644 --- a/code/sync.cpp +++ b/code/sync.cpp @@ -85,6 +85,10 @@ printf("\n Total Computation, TIme = %lf \n", duration_total); printf("\nTask Done!\n"); + + free(cpu_mem, my_gpu_queue); + free(host_mem, my_gpu_queue); + free(device_mem, my_gpu_queue); return 0; }