atomic_auto.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_atomic_auto_HPP
#define RAJA_policy_atomic_auto_HPP
 
#include "RAJA/config.hpp"
 
#include "RAJA/util/macros.hpp"
 
#if !defined(RAJA_ENABLE_DESUL_ATOMICS)
#include "RAJA/policy/sequential/atomic.hpp"
#endif
 
#if defined(__CUDA_ARCH__) && defined(RAJA_CUDA_ACTIVE)
#define RAJA_AUTO_ATOMIC                                                       \
  RAJA::cuda_atomic {}
#elif defined(__HIP_DEVICE_COMPILE__) && defined(RAJA_HIP_ACTIVE)
#define RAJA_AUTO_ATOMIC                                                       \
  RAJA::hip_atomic {}
#elif defined(__SYCL_DEVICE_ONLY__)
#define RAJA_AUTO_ATOMIC                                                       \
  RAJA::sycl_atomic {}
#elif defined(RAJA_ENABLE_OPENMP)
#define RAJA_AUTO_ATOMIC                                                       \
  RAJA::omp_atomic {}
#else
#define RAJA_AUTO_ATOMIC                                                       \
  RAJA::seq_atomic {}
#endif
 
 
namespace RAJA
{
 
struct auto_atomic
{};
 
template<typename T>
RAJA_INLINE RAJA_HOST_DEVICE T atomicLoad(auto_atomic, T* acc)
{
  return atomicLoad(RAJA_AUTO_ATOMIC, acc);
}
 
template<typename T>
RAJA_INLINE RAJA_HOST_DEVICE void atomicStore(auto_atomic, T* acc, T value)
{
  atomicStore(RAJA_AUTO_ATOMIC, acc, value);
}
 
template<typename T>
RAJA_INLINE RAJA_HOST_DEVICE T atomicAdd(auto_atomic, T* acc, T value)
{
  return atomicAdd(RAJA_AUTO_ATOMIC, acc, value);
}
 
template<typename T>
RAJA_INLINE RAJA_HOST_DEVICE T atomicSub(auto_atomic, T* acc, T value)
{
  return atomicSub(RAJA_AUTO_ATOMIC, acc, value);
}
 
template<typename T>
RAJA_INLINE RAJA_HOST_DEVICE T atomicMin(auto_atomic, T* acc, T value)
{
  return atomicMin(RAJA_AUTO_ATOMIC, acc, value);
}
 
template<typename T>
RAJA_INLINE RAJA_HOST_DEVICE T atomicMax(auto_atomic, T* acc, T value)
{
  return atomicMax(RAJA_AUTO_ATOMIC, acc, value);
}
 
template<typename T>
RAJA_INLINE RAJA_HOST_DEVICE T atomicInc(auto_atomic, T* acc)
{
  return atomicInc(RAJA_AUTO_ATOMIC, acc);
}
 
template<typename T>
RAJA_INLINE RAJA_HOST_DEVICE T atomicInc(auto_atomic, T* acc, T compare)
{
  return atomicInc(RAJA_AUTO_ATOMIC, acc, compare);
}
 
template<typename T>
RAJA_INLINE RAJA_HOST_DEVICE T atomicDec(auto_atomic, T* acc)
{
  return atomicDec(RAJA_AUTO_ATOMIC, acc);
}
 
template<typename T>
RAJA_INLINE RAJA_HOST_DEVICE T atomicDec(auto_atomic, T* acc, T compare)
{
  return atomicDec(RAJA_AUTO_ATOMIC, acc, compare);
}
 
template<typename T>
RAJA_INLINE RAJA_HOST_DEVICE T atomicAnd(auto_atomic, T* acc, T value)
{
  return atomicAnd(RAJA_AUTO_ATOMIC, acc, value);
}
 
template<typename T>
RAJA_INLINE RAJA_HOST_DEVICE T atomicOr(auto_atomic, T* acc, T value)
{
  return atomicOr(RAJA_AUTO_ATOMIC, acc, value);
}
 
template<typename T>
RAJA_INLINE RAJA_HOST_DEVICE T atomicXor(auto_atomic, T* acc, T value)
{
  return atomicXor(RAJA_AUTO_ATOMIC, acc, value);
}
 
template<typename T>
RAJA_INLINE RAJA_HOST_DEVICE T atomicExchange(auto_atomic, T* acc, T value)
{
  return atomicExchange(RAJA_AUTO_ATOMIC, acc, value);
}
 
template<typename T>
RAJA_INLINE RAJA_HOST_DEVICE T
atomicCAS(auto_atomic, T* acc, T compare, T value)
{
  return atomicCAS(RAJA_AUTO_ATOMIC, acc, compare, value);
}
 
 
}  // namespace RAJA
 
// make sure this define doesn't bleed out of this header
#undef RAJA_AUTO_ATOMIC
 
#endif