From dcf734387e4de7d037063f22cc5f2496f7c7b209 Mon Sep 17 00:00:00 2001
From: Tom Deakin <tom.deakin@bristol.ac.uk>
Date: Tue, 28 Jul 2015 13:59:31 +0100
Subject: [PATCH] Free host arrays at end of program

---
 cuda-stream.cu | 5 +++++
 ocl-stream.cpp | 5 +++++
 2 files changed, 10 insertions(+)

diff --git a/cuda-stream.cu b/cuda-stream.cu
index 9b06efd..d5eaa31 100644
--- a/cuda-stream.cu
+++ b/cuda-stream.cu
@@ -288,6 +288,11 @@ int main(int argc, char *argv[])
             << std::endl;
     }
 
+    // Free host vectors
+    free(h_a);
+    free(h_b);
+    free(h_c);
+
 }
 
 std::string getDeviceName(int device)
diff --git a/ocl-stream.cpp b/ocl-stream.cpp
index a2f6f2f..e64548a 100644
--- a/ocl-stream.cpp
+++ b/ocl-stream.cpp
@@ -313,6 +313,11 @@ int main(int argc, char *argv[])
                 << std::endl;
         }
 
+        // Free host vectors
+        free(h_a);
+        free(h_b);
+        free(h_c);
+
     }
     catch (cl::Error &e)
     {