libcxx/include/__cxx03/experimental/__simd/traits.h

   1 // -*- C++ -*-
   2 //===----------------------------------------------------------------------===//
   3 //
   4 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   5 // See https://llvm.org/LICENSE.txt for license information.
   6 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 #ifndef _LIBCPP_EXPERIMENTAL___SIMD_TRAITS_H
  11 #define _LIBCPP_EXPERIMENTAL___SIMD_TRAITS_H
  12
  13 #include <__cxx03/__bit/bit_ceil.h>
  14 #include <__cxx03/__type_traits/integral_constant.h>
  15 #include <__cxx03/__type_traits/is_same.h>
  16 #include <__cxx03/cstddef>
  17 #include <__cxx03/experimental/__config>
  18 #include <__cxx03/experimental/__simd/declaration.h>
  19 #include <__cxx03/experimental/__simd/utility.h>
  20
  21 #if _LIBCPP_STD_VER >= 17 && defined(_LIBCPP_ENABLE_EXPERIMENTAL)
  22
  23 _LIBCPP_BEGIN_NAMESPACE_EXPERIMENTAL
  24 inline namespace parallelism_v2 {
  25
  26 // traits [simd.traits]
  27 template <class _Tp>
  28 inline constexpr bool is_abi_tag_v = false;
  29
  30 template <class _Tp>
  31 struct is_abi_tag : bool_constant<is_abi_tag_v<_Tp>> {};
  32
  33 template <class _Tp>
  34 inline constexpr bool is_simd_v = false;
  35
  36 template <class _Tp>
  37 struct is_simd : bool_constant<is_simd_v<_Tp>> {};
  38
  39 template <class _Tp>
  40 inline constexpr bool is_simd_mask_v = false;
  41
  42 template <class _Tp>
  43 struct is_simd_mask : bool_constant<is_simd_mask_v<_Tp>> {};
  44
  45 template <class _Tp>
  46 inline constexpr bool is_simd_flag_type_v = false;
  47
  48 template <class _Tp>
  49 struct is_simd_flag_type : bool_constant<is_simd_flag_type_v<_Tp>> {};
  50
  51 template <class _Tp, class _Abi = simd_abi::compatible<_Tp>, bool = (__is_vectorizable_v<_Tp> && is_abi_tag_v<_Abi>)>
  52 struct simd_size : integral_constant<size_t, _Abi::__simd_size> {};
  53
  54 template <class _Tp, class _Abi>
  55 struct simd_size<_Tp, _Abi, false> {};
  56
  57 template <class _Tp, class _Abi = simd_abi::compatible<_Tp>>
  58 inline constexpr size_t simd_size_v = simd_size<_Tp, _Abi>::value;
  59
  60 template <class _Tp,
  61           class _Up = typename _Tp::value_type,
  62           bool      = (is_simd_v<_Tp> && __is_vectorizable_v<_Up>) || (is_simd_mask_v<_Tp> && is_same_v<_Up, bool>)>
  63 struct memory_alignment : integral_constant<size_t, std::__bit_ceil(sizeof(_Up) * _Tp::size())> {};
  64
  65 template <class _Tp, class _Up>
  66 struct memory_alignment<_Tp, _Up, false> {};
  67
  68 template <class _Tp, class _Up = typename _Tp::value_type>
  69 inline constexpr size_t memory_alignment_v = memory_alignment<_Tp, _Up>::value;
  70
  71 } // namespace parallelism_v2
  72 _LIBCPP_END_NAMESPACE_EXPERIMENTAL
  73
  74 #endif // _LIBCPP_STD_VER >= 17 && defined(_LIBCPP_ENABLE_EXPERIMENTAL)
  75 #endif // _LIBCPP_EXPERIMENTAL___SIMD_TRAITS_H