xref: /freebsd-src/contrib/llvm-project/libcxx/include/experimental/__simd/traits.h (revision cb14a3fe5122c879eae1fb480ed7ce82a699ddb6)
15f757f3fSDimitry Andric // -*- C++ -*-
25f757f3fSDimitry Andric //===----------------------------------------------------------------------===//
35f757f3fSDimitry Andric //
45f757f3fSDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
55f757f3fSDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
65f757f3fSDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
75f757f3fSDimitry Andric //
85f757f3fSDimitry Andric //===----------------------------------------------------------------------===//
95f757f3fSDimitry Andric 
105f757f3fSDimitry Andric #ifndef _LIBCPP_EXPERIMENTAL___SIMD_TRAITS_H
115f757f3fSDimitry Andric #define _LIBCPP_EXPERIMENTAL___SIMD_TRAITS_H
125f757f3fSDimitry Andric 
13*cb14a3feSDimitry Andric #include <__bit/bit_ceil.h>
145f757f3fSDimitry Andric #include <__type_traits/integral_constant.h>
155f757f3fSDimitry Andric #include <__type_traits/is_same.h>
165f757f3fSDimitry Andric #include <cstddef>
175f757f3fSDimitry Andric #include <experimental/__config>
185f757f3fSDimitry Andric #include <experimental/__simd/declaration.h>
195f757f3fSDimitry Andric #include <experimental/__simd/utility.h>
205f757f3fSDimitry Andric 
215f757f3fSDimitry Andric #if _LIBCPP_STD_VER >= 17 && defined(_LIBCPP_ENABLE_EXPERIMENTAL)
225f757f3fSDimitry Andric 
235f757f3fSDimitry Andric _LIBCPP_BEGIN_NAMESPACE_EXPERIMENTAL
245f757f3fSDimitry Andric inline namespace parallelism_v2 {
255f757f3fSDimitry Andric 
265f757f3fSDimitry Andric // traits [simd.traits]
275f757f3fSDimitry Andric template <class _Tp>
285f757f3fSDimitry Andric inline constexpr bool is_abi_tag_v = false;
295f757f3fSDimitry Andric 
305f757f3fSDimitry Andric template <class _Tp>
315f757f3fSDimitry Andric struct is_abi_tag : bool_constant<is_abi_tag_v<_Tp>> {};
325f757f3fSDimitry Andric 
335f757f3fSDimitry Andric template <class _Tp>
345f757f3fSDimitry Andric inline constexpr bool is_simd_v = false;
355f757f3fSDimitry Andric 
365f757f3fSDimitry Andric template <class _Tp>
375f757f3fSDimitry Andric struct is_simd : bool_constant<is_simd_v<_Tp>> {};
385f757f3fSDimitry Andric 
395f757f3fSDimitry Andric template <class _Tp>
405f757f3fSDimitry Andric inline constexpr bool is_simd_mask_v = false;
415f757f3fSDimitry Andric 
425f757f3fSDimitry Andric template <class _Tp>
435f757f3fSDimitry Andric struct is_simd_mask : bool_constant<is_simd_mask_v<_Tp>> {};
445f757f3fSDimitry Andric 
455f757f3fSDimitry Andric template <class _Tp>
465f757f3fSDimitry Andric inline constexpr bool is_simd_flag_type_v = false;
475f757f3fSDimitry Andric 
485f757f3fSDimitry Andric template <class _Tp>
495f757f3fSDimitry Andric struct is_simd_flag_type : bool_constant<is_simd_flag_type_v<_Tp>> {};
505f757f3fSDimitry Andric 
515f757f3fSDimitry Andric template <class _Tp, class _Abi = simd_abi::compatible<_Tp>, bool = (__is_vectorizable_v<_Tp> && is_abi_tag_v<_Abi>)>
525f757f3fSDimitry Andric struct simd_size : integral_constant<size_t, _Abi::__simd_size> {};
535f757f3fSDimitry Andric 
545f757f3fSDimitry Andric template <class _Tp, class _Abi>
555f757f3fSDimitry Andric struct simd_size<_Tp, _Abi, false> {};
565f757f3fSDimitry Andric 
575f757f3fSDimitry Andric template <class _Tp, class _Abi = simd_abi::compatible<_Tp>>
585f757f3fSDimitry Andric inline constexpr size_t simd_size_v = simd_size<_Tp, _Abi>::value;
595f757f3fSDimitry Andric 
605f757f3fSDimitry Andric template <class _Tp,
615f757f3fSDimitry Andric           class _Up = typename _Tp::value_type,
625f757f3fSDimitry Andric           bool      = (is_simd_v<_Tp> && __is_vectorizable_v<_Up>) || (is_simd_mask_v<_Tp> && is_same_v<_Up, bool>)>
63*cb14a3feSDimitry Andric struct memory_alignment : integral_constant<size_t, std::__bit_ceil(sizeof(_Up) * _Tp::size())> {};
645f757f3fSDimitry Andric 
655f757f3fSDimitry Andric template <class _Tp, class _Up>
665f757f3fSDimitry Andric struct memory_alignment<_Tp, _Up, false> {};
675f757f3fSDimitry Andric 
685f757f3fSDimitry Andric template <class _Tp, class _Up = typename _Tp::value_type>
695f757f3fSDimitry Andric inline constexpr size_t memory_alignment_v = memory_alignment<_Tp, _Up>::value;
705f757f3fSDimitry Andric 
715f757f3fSDimitry Andric } // namespace parallelism_v2
725f757f3fSDimitry Andric _LIBCPP_END_NAMESPACE_EXPERIMENTAL
735f757f3fSDimitry Andric 
745f757f3fSDimitry Andric #endif // _LIBCPP_STD_VER >= 17 && defined(_LIBCPP_ENABLE_EXPERIMENTAL)
755f757f3fSDimitry Andric #endif // _LIBCPP_EXPERIMENTAL___SIMD_TRAITS_H
76