SoAArray.hpp

Go to the documentation of this file.

 
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~//
// Copyright (c) Lawrence Livermore National Security, LLC and other
// RAJA Project Developers. See top-level LICENSE and COPYRIGHT
// files for dates and other details. No copyright assignment is required
// to contribute to RAJA.
//
// SPDX-License-Identifier: (BSD-3-Clause)
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~//
 
#ifndef RAJA_SOA_ARRAY_HPP
#define RAJA_SOA_ARRAY_HPP
 
#include "RAJA/config.hpp"
 
// for RAJA::reduce::detail::ValueLoc
#include "RAJA/pattern/detail/reduce.hpp"
 
namespace RAJA
{
 
namespace detail
{
 
template<typename T, size_t size>
class SoAArray
{
  using value_type = T;
 
public:
  constexpr RAJA_HOST_DEVICE value_type get(size_t i) const { return mem[i]; }
 
  constexpr RAJA_HOST_DEVICE void set(size_t i, value_type val)
  {
    mem[i] = val;
  }
 
private:
  value_type mem[size];
};
 
template<typename T, typename IndexType, bool doing_min, size_t size>
class SoAArray<::RAJA::reduce::detail::ValueLoc<T, IndexType, doing_min>, size>
{
  using value_type  = ::RAJA::reduce::detail::ValueLoc<T, IndexType, doing_min>;
  using first_type  = T;
  using second_type = IndexType;
 
public:
  constexpr RAJA_HOST_DEVICE value_type get(size_t i) const
  {
    return value_type(mem[i], mem_idx[i]);
  }
 
  constexpr RAJA_HOST_DEVICE void set(size_t i, value_type val)
  {
    mem[i]     = val;
    mem_idx[i] = val.getLoc();
  }
 
private:
  first_type mem[size];
  second_type mem_idx[size];
};
 
}  // namespace detail
 
}  // namespace RAJA
 
#endif /* RAJA_SOA_ARRAY_HPP */