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

use signed ints for ACC array size

Browse Source
This commit is contained in:
Tom Deakin 2021-01-12 10:14:44 +00:00
parent e6c200a2d3
commit 850c63d69b
2 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
ACCStream.cpp
View File

@ -8,7 +8,7 @@
#include "ACCStream.h" #include "ACCStream.h"
template <class T> template <class T>
ACCStream<T>::ACCStream(const unsigned int ARRAY_SIZE, int device) ACCStream<T>::ACCStream(const int ARRAY_SIZE, int device)
{ {
acc_device_t device_type = acc_get_device_type(); acc_device_t device_type = acc_get_device_type();
acc_set_device_num(device, device_type); acc_set_device_num(device, device_type);
@ -27,7 +27,7 @@ template <class T>
ACCStream<T>::~ACCStream() ACCStream<T>::~ACCStream()
{ {
// End data region on device // End data region on device
unsigned int array_size = this->array_size; int array_size = this->array_size;
#pragma acc exit data delete(a[0:array_size], b[0:array_size], c[0:array_size]) #pragma acc exit data delete(a[0:array_size], b[0:array_size], c[0:array_size])
{} {}
@ -39,7 +39,7 @@ ACCStream<T>::~ACCStream()
template <class T> template <class T>
void ACCStream<T>::init_arrays(T initA, T initB, T initC) void ACCStream<T>::init_arrays(T initA, T initB, T initC)
{ {
unsigned int array_size = this->array_size; int array_size = this->array_size;
T * restrict a = this->a; T * restrict a = this->a;
T * restrict b = this->b; T * restrict b = this->b;
T * restrict c = this->c; T * restrict c = this->c;
@ -65,7 +65,7 @@ void ACCStream<T>::read_arrays(std::vector<T>& h_a, std::vector<T>& h_b, std::ve
template <class T> template <class T>
void ACCStream<T>::copy() void ACCStream<T>::copy()
{ {
unsigned int array_size = this->array_size; int array_size = this->array_size;
T * restrict a = this->a; T * restrict a = this->a;
T * restrict c = this->c; T * restrict c = this->c;
#pragma acc parallel loop present(a[0:array_size], c[0:array_size]) wait #pragma acc parallel loop present(a[0:array_size], c[0:array_size]) wait
@ -80,7 +80,7 @@ void ACCStream<T>::mul()
{ {
const T scalar = startScalar; const T scalar = startScalar;
unsigned int array_size = this->array_size; int array_size = this->array_size;
T * restrict b = this->b; T * restrict b = this->b;
T * restrict c = this->c; T * restrict c = this->c;
#pragma acc parallel loop present(b[0:array_size], c[0:array_size]) wait #pragma acc parallel loop present(b[0:array_size], c[0:array_size]) wait
@ -93,7 +93,7 @@ void ACCStream<T>::mul()
template <class T> template <class T>
void ACCStream<T>::add() void ACCStream<T>::add()
{ {
unsigned int array_size = this->array_size; int array_size = this->array_size;
T * restrict a = this->a; T * restrict a = this->a;
T * restrict b = this->b; T * restrict b = this->b;
T * restrict c = this->c; T * restrict c = this->c;
@ -109,7 +109,7 @@ void ACCStream<T>::triad()
{ {
const T scalar = startScalar; const T scalar = startScalar;
unsigned int array_size = this->array_size; int array_size = this->array_size;
T * restrict a = this->a; T * restrict a = this->a;
T * restrict b = this->b; T * restrict b = this->b;
T * restrict c = this->c; T * restrict c = this->c;
@ -125,7 +125,7 @@ T ACCStream<T>::dot()
{ {
T sum = 0.0; T sum = 0.0;
unsigned int array_size = this->array_size; int array_size = this->array_size;
T * restrict a = this->a; T * restrict a = this->a;
T * restrict b = this->b; T * restrict b = this->b;
#pragma acc parallel loop reduction(+:sum) present(a[0:array_size], b[0:array_size]) wait #pragma acc parallel loop reduction(+:sum) present(a[0:array_size], b[0:array_size]) wait

4
ACCStream.h
View File

@ -21,14 +21,14 @@ class ACCStream : public Stream<T>
{ {
protected: protected:
// Size of arrays // Size of arrays
unsigned int array_size; int array_size;
// Device side pointers // Device side pointers
T *a; T *a;
T *b; T *b;
T *c; T *c;
public: public:
ACCStream(const unsigned int, int); ACCStream(const int, int);
~ACCStream(); ~ACCStream();
virtual void copy() override; virtual void copy() override;