Fix GCC AMD build for OpenMP offload

2020-12-30 14:40:21 +00:00 · 2020-12-30 14:40:21 +00:00 · 4abb080a0e
commit 4abb080a0e
parent cf42335e7a
1 changed files with 3 additions and 0 deletions
--- a/OMPStream.cpp
+++ b/OMPStream.cpp
@ -23,6 +23,9 @@ OMPStream<T>::OMPStream(const unsigned int ARRAY_SIZE, int device)

 #ifdef OMP_TARGET_GPU
  omp_set_default_device(device);
+  T *a = this->a;
+  T *b = this->b;
+  T *c = this->c;
  // Set up data region on device
  #pragma omp target enter data map(alloc: a[0:array_size], b[0:array_size], c[0:array_size])
  {}