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

use signed ints for HIP

Browse Source
This commit is contained in:
Tom Deakin 2021-01-12 10:24:27 +00:00
parent 9a69d3d5d9
commit 66aaec281f
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
HIPStream.cpp
View File

@ -22,7 +22,7 @@ void check_error(void)
}
template <class T>
HIPStream<T>::HIPStream(const unsigned int ARRAY_SIZE, const int device_index)
HIPStream<T>::HIPStream(const int ARRAY_SIZE, const int device_index)
{
// The array size must be divisible by TBSIZE for kernel launches
@ -183,7 +183,7 @@ void HIPStream<T>::triad()
}
template <class T>
__global__ void dot_kernel(const T * a, const T * b, T * sum, unsigned int array_size)
__global__ void dot_kernel(const T * a, const T * b, T * sum, int array_size)
{
__shared__ T tb_sum[TBSIZE];

4
HIPStream.h
View File

@ -20,7 +20,7 @@ class HIPStream : public Stream<T>
{
protected:
// Size of arrays
unsigned int array_size;
int array_size;
// Host array for partial sums for dot kernel
T *sums;
@ -34,7 +34,7 @@ class HIPStream : public Stream<T>
public:
HIPStream(const unsigned int, const int);
HIPStream(const int, const int);
~HIPStream();
virtual void copy() override;