From 6424aacec6dca3cdc29f5ebb5497086c8b8b34ed Mon Sep 17 00:00:00 2001
From: PatricZhao <patric.zhao@intel.com>
Date: Fri, 22 Oct 2021 20:32:34 +0800
Subject: [PATCH] Update data_movement_ex.cpp

---
 code/data_movement_ex.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/code/data_movement_ex.cpp b/code/data_movement_ex.cpp
index 6d45201..c523f40 100644
--- a/code/data_movement_ex.cpp
+++ b/code/data_movement_ex.cpp
@@ -35,6 +35,9 @@
         printf("%d, ", host_mem[i]);
      }
      printf("\nTask Done!\n");
+      
+     free(host_mem, my_gpu_queue);
+     free(device_mem, my_gpu_queue);  
    
      return 0;
    }