cory-master/BabelStream
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add map clauses to OpenMP 4.5 kernels

Browse Source
This commit is contained in:
Tom Deakin 2016-05-11 15:17:06 +01:00
parent eb10c716f2
commit 4954ef7cf0
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
OMP45Stream.cpp
View File

@ -12,7 +12,7 @@ OMP45Stream<T>::OMP45Stream(const unsigned int ARRAY_SIZE, T *a, T *b, T *c, int
this->a = a;
this->b = b;
this->c = c;
#pragma omp target enter data map(alloc: a[0:array_size], b[0:array_size], c[0:array_size])
#pragma omp target enter data map(to: a[0:array_size], b[0:array_size], c[0:array_size])
{}
}
@ -54,7 +54,7 @@ void OMP45Stream<T>::copy()
unsigned int array_size = this->array_size;
T *a = this->a;
T *c = this->c;
#pragma omp target teams distribute parallel for simd
#pragma omp target teams distribute parallel for simd map(to: a[0:array_size], c[0:array_size])
for (int i = 0; i < array_size; i++)
{
c[i] = a[i];
@ -69,7 +69,7 @@ void OMP45Stream<T>::mul()
unsigned int array_size = this->array_size;
T *b = this->b;
T *c = this->c;
#pragma omp target teams distribute parallel for simd
#pragma omp target teams distribute parallel for simd map(to: b[0:array_size], c[0:array_size])
for (int i = 0; i < array_size; i++)
{
b[i] = scalar * c[i];
@ -83,7 +83,7 @@ void OMP45Stream<T>::add()
T *a = this->a;
T *b = this->b;
T *c = this->c;
#pragma omp target teams distribute parallel for simd
#pragma omp target teams distribute parallel for simd map(to: a[0:array_size], b[0:array_size], c[0:array_size])
for (int i = 0; i < array_size; i++)
{
c[i] = a[i] + b[i];
@ -99,7 +99,7 @@ void OMP45Stream<T>::triad()
T *a = this->a;
T *b = this->b;
T *c = this->c;
#pragma omp target teams distribute parallel for simd
#pragma omp target teams distribute parallel for simd map(to: a[0:array_size], b[0:array_size], c[0:array_size])
for (int i = 0; i < array_size; i++)
{
a[i] = b[i] + scalar * c[i];