privatizer.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_PRIVATIZER_HPP
#define __RAJA_PRIVATIZER_HPP
 
#include "RAJA/config.hpp"
#include "camp/camp.hpp"
#include "camp/concepts.hpp"
 
namespace RAJA
{
 
namespace internal
{
 
// template <typename T>
// struct HasPrivatizer : DefineConcept(typename T::privatizer(camp::val<T>()))
// {
// };
// DefineTypeTraitFromConcept(has_privatizer, HasPrivatizer);
 
template<typename T>
class has_privatizer
{
private:
  template<typename C>
  static auto Test(void*)
      -> decltype(camp::val<typename C::privatizer>(), camp::true_type {});
 
  template<typename>
  static camp::false_type Test(...);
 
public:
  static bool const value = decltype(Test<T>(0))::value;
};
 
static_assert(!has_privatizer<int>::value, "if this fires, abandon all hope");
 
struct GenericWrapperBase
{};
 
template<typename T>
struct Privatizer
{
  using value_type     = camp::decay<T>;
  using reference_type = value_type&;
  value_type priv;
  static_assert(!has_privatizer<T>::value,
                "Privatizer selected inappropriately, this is almost certainly "
                "a bug");
  static_assert(!std::is_base_of<GenericWrapperBase, T>::value,
                "Privatizer selected inappropriately, this is almost certainly "
                "a bug");
 
  RAJA_SUPPRESS_HD_WARN
  RAJA_HOST_DEVICE Privatizer(const T& o) : priv {o} {}
 
  RAJA_SUPPRESS_HD_WARN
  RAJA_HOST_DEVICE reference_type get_priv() { return priv; }
};
 
template<typename T,
         typename std::enable_if<!has_privatizer<T>::value>::type* = nullptr>
RAJA_HOST_DEVICE auto thread_privatize(const T& item) -> Privatizer<T>
{
  return Privatizer<T> {item};
}
 
RAJA_SUPPRESS_HD_WARN
template<typename T,
         typename std::enable_if<has_privatizer<T>::value>::type* = nullptr>
RAJA_HOST_DEVICE auto thread_privatize(const T& item) -> typename T::privatizer
{
  return typename T::privatizer {item};
}
 
}  // namespace internal
 
}  // namespace RAJA
 
#endif /* __RAJA_PRIVATIZER_HPP */