From d75b0004de9a1fc071af2eea37b37772e2d3cce8 Mon Sep 17 00:00:00 2001
From: Tom Deakin <tom.deakin@bristol.ac.uk>
Date: Tue, 13 Dec 2016 11:45:30 +0000
Subject: [PATCH] [OMP] Update deconstructor to only call target region if
 building for GPU

---
 OMPStream.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/OMPStream.cpp b/OMPStream.cpp
index da51937..189cacb 100644
--- a/OMPStream.cpp
+++ b/OMPStream.cpp
@@ -31,6 +31,7 @@ OMPStream<T>::OMPStream(const unsigned int ARRAY_SIZE, T *a, T *b, T *c, int dev
 template <class T>
 OMPStream<T>::~OMPStream()
 {
+#ifdef OMP_TARGET_GPU
   // End data region on device
   unsigned int array_size = this->array_size;
   T *a = this->a;
@@ -38,6 +39,11 @@ OMPStream<T>::~OMPStream()
   T *c = this->c;
   #pragma omp target exit data map(release: a[0:array_size], b[0:array_size], c[0:array_size])
   {}
+#else
+  free(a);
+  free(b);
+  free(c);
+#endif
 }
 
 template <class T>