experimental/__simd/simd_mask.h

0001 // -*- C++ -*-
0002 //===----------------------------------------------------------------------===//
0003 //
0004 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
0005 // See https://llvm.org/LICENSE.txt for license information.
0006 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
0007 //
0008 //===----------------------------------------------------------------------===//
0009
0010 #ifndef _LIBCPP_EXPERIMENTAL___SIMD_SIMD_MASK_H
0011 #define _LIBCPP_EXPERIMENTAL___SIMD_SIMD_MASK_H
0012
0013 #include <__config>
0014 #include <__cstddef/size_t.h>
0015 #include <__type_traits/enable_if.h>
0016 #include <__type_traits/is_same.h>
0017 #include <experimental/__simd/declaration.h>
0018 #include <experimental/__simd/reference.h>
0019 #include <experimental/__simd/traits.h>
0020
0021 #if _LIBCPP_STD_VER >= 17 && defined(_LIBCPP_ENABLE_EXPERIMENTAL)
0022
0023 _LIBCPP_BEGIN_NAMESPACE_EXPERIMENTAL
0024 inline namespace parallelism_v2 {
0025
0026 // class template simd_mask [simd.mask.class]
0027 // TODO: implement simd_mask class
0028 template <class _Tp, class _Abi>
0029 class simd_mask {
0030   using _Impl _LIBCPP_NODEBUG    = __mask_operations<_Tp, _Abi>;
0031   using _Storage _LIBCPP_NODEBUG = typename _Impl::_MaskStorage;
0032
0033   _Storage __s_;
0034
0035 public:
0036   using value_type = bool;
0037   using reference  = __simd_reference<_Tp, _Storage, value_type>;
0038   using simd_type  = simd<_Tp, _Abi>;
0039   using abi_type   = _Abi;
0040
0041   static _LIBCPP_HIDE_FROM_ABI constexpr size_t size() noexcept { return simd_type::size(); }
0042
0043   _LIBCPP_HIDE_FROM_ABI simd_mask() noexcept = default;
0044
0045   // explicit conversion from and to implementation-defined types
0046   struct __storage_tag_t {};
0047   static constexpr __storage_tag_t __storage_tag{};
0048   explicit _LIBCPP_HIDE_FROM_ABI operator _Storage() const { return __s_; }
0049   explicit _LIBCPP_HIDE_FROM_ABI simd_mask(const _Storage& __s, __storage_tag_t) : __s_(__s) {}
0050
0051   // broadcast constructor
0052   _LIBCPP_HIDE_FROM_ABI explicit simd_mask(value_type __v) noexcept : __s_(_Impl::__broadcast(__v)) {}
0053
0054   // implicit type conversion constructor
0055   template <class _Up, enable_if_t<!is_same_v<_Up, _Tp> && is_same_v<abi_type, simd_abi::fixed_size<size()>>, int> = 0>
0056   _LIBCPP_HIDE_FROM_ABI simd_mask(const simd_mask<_Up, simd_abi::fixed_size<size()>>& __v) noexcept {
0057     for (size_t __i = 0; __i < size(); __i++) {
0058       (*this)[__i] = __v[__i];
0059     }
0060   }
0061
0062   // load constructor
0063   template <class _Flags, enable_if_t<is_simd_flag_type_v<_Flags>, int> = 0>
0064   _LIBCPP_HIDE_FROM_ABI simd_mask(const value_type* __mem, _Flags) {
0065     _Impl::__load(__s_, _Flags::template __apply<simd_mask>(__mem));
0066   }
0067
0068   // copy functions
0069   template <class _Flags, enable_if_t<is_simd_flag_type_v<_Flags>, int> = 0>
0070   _LIBCPP_HIDE_FROM_ABI void copy_from(const value_type* __mem, _Flags) {
0071     _Impl::__load(__s_, _Flags::template __apply<simd_mask>(__mem));
0072   }
0073
0074   template <class _Flags, enable_if_t<is_simd_flag_type_v<_Flags>, int> = 0>
0075   _LIBCPP_HIDE_FROM_ABI void copy_to(value_type* __mem, _Flags) const {
0076     _Impl::__store(__s_, _Flags::template __apply<simd_mask>(__mem));
0077   }
0078
0079   // scalar access [simd.mask.subscr]
0080   _LIBCPP_HIDE_FROM_ABI reference operator[](size_t __i) noexcept { return reference(__s_, __i); }
0081   _LIBCPP_HIDE_FROM_ABI value_type operator[](size_t __i) const noexcept { return __s_.__get(__i); }
0082 };
0083
0084 template <class _Tp, class _Abi>
0085 inline constexpr bool is_simd_mask_v<simd_mask<_Tp, _Abi>> = true;
0086
0087 template <class _Tp>
0088 using native_simd_mask = simd_mask<_Tp, simd_abi::native<_Tp>>;
0089
0090 template <class _Tp, int _Np>
0091 using fixed_size_simd_mask = simd_mask<_Tp, simd_abi::fixed_size<_Np>>;
0092
0093 } // namespace parallelism_v2
0094 _LIBCPP_END_NAMESPACE_EXPERIMENTAL
0095
0096 #endif // _LIBCPP_STD_VER >= 17 && defined(_LIBCPP_ENABLE_EXPERIMENTAL)
0097 #endif // _LIBCPP_EXPERIMENTAL___SIMD_SIMD_MASK_H