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> template <class T>
KokkosStream<T>::KokkosStream( KokkosStream<T>::KokkosStream(
const unsigned int ARRAY_SIZE, const int device_index) const int ARRAY_SIZE, const int device_index)
: array_size(ARRAY_SIZE) : array_size(ARRAY_SIZE)
{ {
Kokkos::initialize(); Kokkos::initialize();

4
KokkosStream.hpp
View File

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