scalar.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_policy_vector_register_scalar_HPP
#define RAJA_policy_vector_register_scalar_HPP
 
#include "RAJA/pattern/tensor/internal/RegisterBase.hpp"
 
namespace RAJA
{
namespace expt
{
 
template<typename T>
class Register<T, scalar_register>
    : public internal::expt::RegisterBase<Register<T, scalar_register>>
{
public:
  using base_type = internal::expt::RegisterBase<Register<T, scalar_register>>;
 
  using register_policy = scalar_register;
  using self_type       = Register<T, scalar_register>;
  using element_type    = T;
  using register_type   = T;
 
  using int_vector_type =
      Register<typename internal::expt::RegisterTraits<scalar_register,
                                                       T>::int_element_type,
               scalar_register>;
 
 
private:
  T m_value;
 
public:
  static constexpr camp::idx_t s_num_elem = 1;
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  constexpr Register() : base_type(), m_value(0) {}
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  constexpr Register(element_type const& c) : base_type(), m_value(c) {}
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  constexpr Register(self_type const& c) : base_type(), m_value(c.m_value) {}
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  self_type& operator=(self_type const& c)
  {
    m_value = c.m_value;
    return *this;
  }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  self_type& load_packed(element_type const* ptr)
  {
    m_value = ptr[0];
    return *this;
  }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  self_type& load_packed_n(element_type const* ptr, camp::idx_t N)
  {
    if (N > 0)
    {
      m_value = ptr[0];
    }
    else
    {
      m_value = element_type(0);
    }
    return *this;
  }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  self_type& load_strided(element_type const* ptr, camp::idx_t)
  {
    m_value = ptr[0];
    return *this;
  }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  self_type& load_strided_n(element_type const* ptr, camp::idx_t, camp::idx_t N)
  {
    if (N > 0)
    {
      m_value = ptr[0];
    }
    else
    {
      m_value = element_type(0);
    }
    return *this;
  }
 
  RAJA_INLINE
  self_type& gather(element_type const* ptr, int_vector_type offsets)
  {
 
    m_value = ptr[offsets.get(0)];
 
    return *this;
  }
 
  RAJA_INLINE
  self_type& gather_n(element_type const* ptr,
                      int_vector_type offsets,
                      camp::idx_t N)
  {
    if (N > 0)
    {
      m_value = ptr[offsets.get(0)];
    }
    else
    {
      m_value = element_type(0);
    }
    return *this;
  }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  self_type const& store_packed(element_type* ptr) const
  {
    ptr[0] = m_value;
    return *this;
  }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  self_type const& store_packed_n(element_type* ptr, camp::idx_t N) const
  {
    if (N > 0)
    {
      ptr[0] = m_value;
    }
    return *this;
  }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  self_type const& store_strided(element_type* ptr, camp::idx_t) const
  {
    ptr[0] = m_value;
    return *this;
  }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  self_type const& store_strided_n(element_type* ptr,
                                   camp::idx_t,
                                   camp::idx_t N) const
  {
    if (N > 0)
    {
      ptr[0] = m_value;
    }
    return *this;
  }
 
  RAJA_INLINE
  self_type const& scatter(element_type* ptr, int_vector_type offsets) const
  {
 
    ptr[offsets.get(0)] = m_value;
 
    return *this;
  }
 
  RAJA_INLINE
  self_type const& scatter_n(element_type* ptr,
                             int_vector_type offsets,
                             camp::idx_t N) const
  {
    if (N > 0)
    {
      ptr[offsets.get(0)] = m_value;
    }
    return *this;
  }
 
  constexpr RAJA_INLINE RAJA_HOST_DEVICE element_type get(camp::idx_t) const
  {
    return m_value;
  }
 
  RAJA_INLINE
 
  RAJA_HOST_DEVICE
  self_type& set(element_type value, camp::idx_t)
  {
    m_value = value;
    return *this;
  }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  self_type& broadcast(element_type const& a)
  {
    m_value = a;
    return *this;
  }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  self_type& copy(self_type const& src)
  {
    m_value = src.m_value;
    return *this;
  }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  self_type add(self_type const& b) const
  {
    return self_type(m_value + b.m_value);
  }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  self_type subtract(self_type const& b) const
  {
    return self_type(m_value - b.m_value);
  }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  self_type multiply(self_type const& b) const
  {
    return self_type(m_value * b.m_value);
  }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  self_type divide(self_type const& b) const
  {
    return self_type(m_value / b.m_value);
  }
 
  RAJA_INLINE
 
  RAJA_HOST_DEVICE
  self_type multiply_add(self_type const& b, self_type const& c) const
  {
    return m_value * b.m_value + c.m_value;
  }
 
  RAJA_INLINE
 
  RAJA_HOST_DEVICE
  self_type multiply_subtract(self_type const& b, self_type const& c) const
  {
    return m_value * b.m_value - c.m_value;
  }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  constexpr element_type sum() const { return m_value; }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  constexpr element_type dot(self_type const& b) const
  {
    return m_value * b.m_value;
  }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  constexpr element_type max() const { return m_value; }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  element_type max_n(camp::idx_t N) const
  {
    return N ? m_value : RAJA::operators::limits<element_type>::min();
    ;
  }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  self_type vmax(self_type a) const
  {
    return self_type(RAJA::max<element_type>(m_value, a.m_value));
  }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  element_type min() const { return m_value; }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  element_type min_n(camp::idx_t N) const
  {
    return N ? m_value : RAJA::operators::limits<element_type>::max();
    ;
  }
 
  RAJA_HOST_DEVICE
 
  RAJA_INLINE
  self_type vmin(self_type a) const
  {
    return self_type(RAJA::min<element_type>(m_value, a.m_value));
  }
};
}  // namespace expt
}  // namespace RAJA
 
 
#endif