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

use signed int for Kokkos array size

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

2
KokkosStream.cpp
View File

@ -9,7 +9,7 @@
template <class T>
KokkosStream<T>::KokkosStream(
const unsigned int ARRAY_SIZE, const int device_index)
const int ARRAY_SIZE, const int device_index)
: array_size(ARRAY_SIZE)
{
Kokkos::initialize();

4
KokkosStream.hpp
View File

@ -22,7 +22,7 @@ class KokkosStream : public Stream<T>
{
protected:
// Size of arrays
unsigned int array_size;
int array_size;
// Device side pointers to arrays
typename Kokkos::View<T*>* d_a;
@ -34,7 +34,7 @@ class KokkosStream : public Stream<T>
public:
KokkosStream(const unsigned int, const int);
KokkosStream(const int, const int);
~KokkosStream();
virtual void copy() override;