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

Update OpenACC for Issue #80

Browse Source
This commit is contained in:
Tom Deakin 2020-12-07 11:50:20 +00:00
parent b00120d346
commit 5a93022fc1
3 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
ACCStream.cpp
View File

@ -8,7 +8,7 @@
#include "ACCStream.h"
template <class T>
ACCStream<T>::ACCStream(const unsigned int ARRAY_SIZE, T *a, T *b, T *c, int device)
ACCStream<T>::ACCStream(const unsigned int ARRAY_SIZE, int device)
{
acc_device_t device_type = acc_get_device_type();
acc_set_device_num(device, device_type);
@ -16,9 +16,9 @@ ACCStream<T>::ACCStream(const unsigned int ARRAY_SIZE, T *a, T *b, T *c, int dev
array_size = ARRAY_SIZE;
// Set up data region on device
this->a = a;
this->b = b;
this->c = c;
a = new T[array_size];
b = new T[array_size];
c = new T[array_size];
#pragma acc enter data create(a[0:array_size], b[0:array_size], c[0:array_size])
{}
}
@ -28,11 +28,12 @@ ACCStream<T>::~ACCStream()
{
// End data region on device
unsigned int array_size = this->array_size;
T *a = this->a;
T *b = this->b;
T *c = this->c;
#pragma acc exit data delete(a[0:array_size], b[0:array_size], c[0:array_size])
{}
delete[] a;
delete[] b;
delete[] c;
}
template <class T>

2
ACCStream.h
View File

@ -28,7 +28,7 @@ class ACCStream : public Stream<T>
T *c;
public:
ACCStream(const unsigned int, T*, T*, T*, int);
ACCStream(const unsigned int, int);
~ACCStream();
virtual void copy() override;

4
main.cpp
View File

@ -162,7 +162,7 @@ void run()
#elif defined(ACC)
// Use the OpenACC implementation
stream = new ACCStream<T>(ARRAY_SIZE, a.data(), b.data(), c.data(), deviceIndex);
stream = new ACCStream<T>(ARRAY_SIZE, deviceIndex);
#elif defined(SYCL)
// Use the SYCL implementation
@ -362,7 +362,7 @@ void run_triad()
#elif defined(ACC)
// Use the OpenACC implementation
stream = new ACCStream<T>(ARRAY_SIZE, a.data(), b.data(), c.data(), deviceIndex);
stream = new ACCStream<T>(ARRAY_SIZE, deviceIndex);
#elif defined(STD)
// Use the STD implementation