1 files changed, 12 insertions, 0 deletions
diff --git a/HST-S/baselines/cpu/app_baseline.c b/HST-S/baselines/cpu/app_baseline.c
index 875ecbd..0c766e0 100644
--- a/HST-S/baselines/cpu/app_baseline.c
+++ b/HST-S/baselines/cpu/app_baseline.c
@@ -302,5 +302,17 @@ int main(int argc, char **argv) {
         input_size / timer.time[0]);
     printall(&timer, 0);
 
+#if NUMA
+    numa_free(A, input_size * sizeof(T));
+    if (!p.exp) {
+        numa_free(histo_host, nr_of_dpus * p.bins * sizeof(unsigned int));
+    } else {
+        numa_free(histo_host, p.bins * sizeof(unsigned int));
+    }
+#else
+    free(A);
+    free(histo_host);
+#endif
+
     return 0;
 }