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