arch.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_tensor_arch_HPP
#define RAJA_policy_tensor_arch_HPP
 
#include "RAJA/config.hpp"
 
namespace RAJA
{
 
namespace internal
{
 
namespace expt
{
 
 
template<typename REGISTER_POLICY, typename T>
struct RegisterTraits;
/*
 * using element_type = T;
 * using register_policy = REGISTER_POLICY;
 * static constexpr camp::idx s_num_bits = X;
 * static constexpr camp::idx s_num_elem = Y;
 *
 */
}  // namespace expt
}  // namespace internal
 
//
//
// SIMD register types and policies
//
//
namespace expt
{
 
#ifdef __AVX512F__
struct avx512_register
{};
 
#ifndef RAJA_TENSOR_REGISTER_TYPE
#define RAJA_TENSOR_REGISTER_TYPE RAJA::expt::avx512_register
#endif
#endif
 
 
#ifdef __AVX2__
struct avx2_register
{};
 
#ifndef RAJA_TENSOR_REGISTER_TYPE
#define RAJA_TENSOR_REGISTER_TYPE RAJA::expt::avx2_register
#endif
#endif
 
 
#ifdef __AVX__
struct avx_register
{};
 
#ifndef RAJA_TENSOR_REGISTER_TYPE
#define RAJA_TENSOR_REGISTER_TYPE RAJA::expt::avx_register
#endif
#endif
 
 
#ifdef RAJA_ENABLE_CUDA
 
struct cuda_warp_register
{};
 
#endif
 
 
#ifdef RAJA_ENABLE_HIP
 
struct hip_wave_register
{};
 
#endif
 
// The scalar register is always supported (doesn't require any SIMD/SIMT)
struct scalar_register
{};
 
#ifndef RAJA_TENSOR_REGISTER_TYPE
#define RAJA_TENSOR_REGISTER_TYPE RAJA::expt::scalar_register
 
#endif
 
 
// This sets the default SIMD register that will be used
using default_register = RAJA_TENSOR_REGISTER_TYPE;
 
 
}  // namespace expt
}  // namespace RAJA
 
//
// Now include all of the traits files
//
 
#ifdef __AVX512F__
#include "RAJA/policy/tensor/arch/avx512/traits.hpp"
#endif
 
 
#ifdef __AVX2__
#include "RAJA/policy/tensor/arch/avx2/traits.hpp"
#endif
 
#ifdef __AVX__
#include "RAJA/policy/tensor/arch/avx/traits.hpp"
#endif
 
 
#ifdef RAJA_CUDA_ACTIVE
#include "RAJA/policy/tensor/arch/cuda/traits.hpp"
#endif
 
#ifdef RAJA_HIP_ACTIVE
#include "RAJA/policy/tensor/arch/hip/traits.hpp"
#endif
 
#include "RAJA/policy/tensor/arch/scalar/traits.hpp"
 
 
#endif