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

Add RAJA dot kernel

Browse Source
This commit is contained in:
Tom Deakin 2016-10-24 11:34:40 +01:00
parent 823e12708f
commit 7408ab0366
2 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
RAJAStream.cpp
View File

@ -109,6 +109,23 @@ void RAJAStream<T>::triad()
});
}
template <class T>
T RAJAStream<T>::dot()
{
T* a = d_a;
T* b = d_b;
RAJA::ReduceSum<reduce_policy, T> sum(0.0);
forall<policy>(index_set, [=] RAJA_DEVICE (int index)
{
sum += a[index] * b[index];
});
return T(sum);
}
void listDevices(void)
{
std::cout << "This is not the device you are looking for.";

3
RAJAStream.hpp
View File

@ -18,11 +18,13 @@
typedef RAJA::IndexSet::ExecPolicy<
RAJA::seq_segit,
RAJA::omp_parallel_for_exec> policy;
typedef RAJA::omp_reduce reduce_policy;
#else
const size_t block_size = 128;
typedef RAJA::IndexSet::ExecPolicy<
RAJA::seq_segit,
RAJA::cuda_exec<block_size>> policy;
typedef RAJA::cuda_reduce<block_size> reduce_policy;
#endif
template <class T>
@ -49,6 +51,7 @@ class RAJAStream : public Stream<T>
virtual void add() override;
virtual void mul() override;
virtual void triad() override;
virtual T dot() override;
virtual void write_arrays(
const std::vector<T>& a, const std::vector<T>& b, const std::vector<T>& c) override;