master
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
24inline namespace parallelism_v2 {
25
26// class template simd_mask [simd.mask.class]
27// TODO: implement simd_mask class
28template <class _Tp, class _Abi>
29class simd_mask {
30 using _Impl _LIBCPP_NODEBUG = __mask_operations<_Tp, _Abi>;
31 using _Storage _LIBCPP_NODEBUG = typename _Impl::_MaskStorage;
32
33 _Storage __s_;
34
35public:
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
84template <class _Tp, class _Abi>
85inline constexpr bool is_simd_mask_v<simd_mask<_Tp, _Abi>> = true;
86
87template <class _Tp>
88using native_simd_mask = simd_mask<_Tp, simd_abi::native<_Tp>>;
89
90template <class _Tp, int _Np>
91using 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