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