1 files changed, 10 insertions, 4 deletions
diff --git a/GEMV/baselines/cpu/gemv_openmp.c b/GEMV/baselines/cpu/gemv_openmp.c
index ed7b794..798f7de 100644
--- a/GEMV/baselines/cpu/gemv_openmp.c
+++ b/GEMV/baselines/cpu/gemv_openmp.c
@@ -2,14 +2,14 @@
 #include <stdio.h>
 #include "../../support/timer.h"
 
-#if NUMA
-#include <numaif.h>
-#include <numa.h>
-
 #ifndef T
 #define T double
 #endif
 
+#if NUMA
+#include <numaif.h>
+#include <numa.h>
+
 struct bitmask* bitmask_in;
 struct bitmask* bitmask_out;
 
@@ -70,6 +70,12 @@ int main(int argc, char *argv[])
 
     allocate_dense(rows, cols, &A);
 
+#if NUMA
+    if (bitmask_in) {
+        numa_free_nodemask(bitmask_in);
+    }
+#endif
+
     make_hilbert_mat(rows,cols, &A);
 
 #if NUMA