libcxx/include/experimental/__simd/simd_mask.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_SIMD_MASK_H
  11 #define _LIBCPP_EXPERIMENTAL___SIMD_SIMD_MASK_H
  12
  13 #include <__config>
  14 #include <__cstddef/size_t.h>
  15 #include <__type_traits/enable_if.h>
  16 #include <__type_traits/is_same.h>
  17 #include <experimental/__simd/declaration.h>
  18 #include <experimental/__simd/reference.h>
  19 #include <experimental/__simd/traits.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 // class template simd_mask [simd.mask.class]
  27 // TODO: implement simd_mask class
  28 template <class _Tp, class _Abi>
  29 class simd_mask {
  30   using _Impl _LIBCPP_NODEBUG    = __mask_operations<_Tp, _Abi>;
  31   using _Storage _LIBCPP_NODEBUG = typename _Impl::_MaskStorage;
  32
  33   _Storage __s_;
  34
  35 public:
  36   using value_type = bool;
  37   using reference  = __simd_reference<_Tp, _Storage, value_type>;
  38   using simd_type  = simd<_Tp, _Abi>;
  39   using abi_type   = _Abi;
  40
  41   static _LIBCPP_HIDE_FROM_ABI constexpr size_t size() noexcept { return simd_type::size(); }
  42
  43   _LIBCPP_HIDE_FROM_ABI simd_mask() noexcept = default;
  44
  45   // explicit conversion from and to implementation-defined types
  46   struct __storage_tag_t {};
  47   static constexpr __storage_tag_t __storage_tag{};
  48   explicit _LIBCPP_HIDE_FROM_ABI operator _Storage() const { return __s_; }
  49   explicit _LIBCPP_HIDE_FROM_ABI simd_mask(const _Storage& __s, __storage_tag_t) : __s_(__s) {}
  50
  51   // broadcast constructor
  52   _LIBCPP_HIDE_FROM_ABI explicit simd_mask(value_type __v) noexcept : __s_(_Impl::__broadcast(__v)) {}
  53
  54   // implicit type conversion constructor
  55   template <class _Up, enable_if_t<!is_same_v<_Up, _Tp> && is_same_v<abi_type, simd_abi::fixed_size<size()>>, int> = 0>
  56   _LIBCPP_HIDE_FROM_ABI simd_mask(const simd_mask<_Up, simd_abi::fixed_size<size()>>& __v) noexcept {
  57     for (size_t __i = 0; __i < size(); __i++) {
  58       (*this)[__i] = __v[__i];
  59     }
  60   }
  61
  62   // load constructor
  63   template <class _Flags, enable_if_t<is_simd_flag_type_v<_Flags>, int> = 0>
  64   _LIBCPP_HIDE_FROM_ABI simd_mask(const value_type* __mem, _Flags) {
  65     _Impl::__load(__s_, _Flags::template __apply<simd_mask>(__mem));
  66   }
  67
  68   // copy functions
  69   template <class _Flags, enable_if_t<is_simd_flag_type_v<_Flags>, int> = 0>
  70   _LIBCPP_HIDE_FROM_ABI void copy_from(const value_type* __mem, _Flags) {
  71     _Impl::__load(__s_, _Flags::template __apply<simd_mask>(__mem));
  72   }
  73
  74   template <class _Flags, enable_if_t<is_simd_flag_type_v<_Flags>, int> = 0>
  75   _LIBCPP_HIDE_FROM_ABI void copy_to(value_type* __mem, _Flags) const {
  76     _Impl::__store(__s_, _Flags::template __apply<simd_mask>(__mem));
  77   }
  78
  79   // scalar access [simd.mask.subscr]
  80   _LIBCPP_HIDE_FROM_ABI reference operator[](size_t __i) noexcept { return reference(__s_, __i); }
  81   _LIBCPP_HIDE_FROM_ABI value_type operator[](size_t __i) const noexcept { return __s_.__get(__i); }
  82 };
  83
  84 template <class _Tp, class _Abi>
  85 inline constexpr bool is_simd_mask_v<simd_mask<_Tp, _Abi>> = true;
  86
  87 template <class _Tp>
  88 using native_simd_mask = simd_mask<_Tp, simd_abi::native<_Tp>>;
  89
  90 template <class _Tp, int _Np>
  91 using fixed_size_simd_mask = simd_mask<_Tp, simd_abi::fixed_size<_Np>>;
  92
  93 } // namespace parallelism_v2
  94 _LIBCPP_END_NAMESPACE_EXPERIMENTAL
  95
  96 #endif // _LIBCPP_STD_VER >= 17 && defined(_LIBCPP_ENABLE_EXPERIMENTAL)
  97 #endif // _LIBCPP_EXPERIMENTAL___SIMD_SIMD_MASK_H