25#ifndef _GLIBCXX_SIMD_VEC_H
26#define _GLIBCXX_SIMD_VEC_H 1
29#pragma GCC system_header
32#if __cplusplus >= 202400L
35#include "simd_flags.h"
42#pragma GCC diagnostic push
43#pragma GCC diagnostic ignored "-Wpsabi"
45namespace std _GLIBCXX_VISIBILITY(default)
47_GLIBCXX_BEGIN_NAMESPACE_VERSION
51 template <
typename _Tp,
typename _Ap>
55 using value_type = _Tp;
59 using mask_type = basic_mask<0, void>;
61#define _GLIBCXX_DELETE_SIMD "This specialization is disabled because of an invalid combination " \
62 "of template arguments to basic_vec."
64 basic_vec() =
delete(_GLIBCXX_DELETE_SIMD);
66 ~basic_vec() =
delete(_GLIBCXX_DELETE_SIMD);
68 basic_vec(
const basic_vec&) =
delete(_GLIBCXX_DELETE_SIMD);
70 basic_vec& operator=(
const basic_vec&) =
delete(_GLIBCXX_DELETE_SIMD);
72#undef _GLIBCXX_DELETE_SIMD
75 template <
typename _Tp,
typename _Ap>
78 using _Vp = basic_vec<_Tp, _Ap>;
81 using value_type = _Tp;
85 using mask_type = basic_mask<
sizeof(_Tp), abi_type>;
87 using iterator = __iterator<_Vp>;
89 using const_iterator = __iterator<const _Vp>;
93 {
return {
static_cast<_Vp&
>(*this), 0}; }
95 constexpr const_iterator
96 begin() const noexcept
99 constexpr const_iterator
100 cbegin() const noexcept
101 {
return {
static_cast<const _Vp&
>(*this), 0}; }
103 constexpr default_sentinel_t
107 constexpr default_sentinel_t
108 cend() const noexcept
111 static constexpr auto size = __simd_size_c<_Ap::_S_size>;
113 _VecBase() =
default;
116 template <
typename _Up,
typename _UAbi>
117 requires (_Ap::_S_size != _UAbi::_S_size)
118 _VecBase(
const basic_vec<_Up, _UAbi>&) =
delete(
"size mismatch");
120 template <
typename _Up,
typename _UAbi>
121 requires (_Ap::_S_size == _UAbi::_S_size) && (!__explicitly_convertible_to<_Up, _Tp>)
123 _VecBase(
const basic_vec<_Up, _UAbi>&)
124 =
delete(
"the value types are not convertible");
126 [[__gnu__::__always_inline__]]
128 operator+(
const _Vp& __x,
const _Vp& __y)
noexcept
135 [[__gnu__::__always_inline__]]
137 operator-(
const _Vp& __x,
const _Vp& __y)
noexcept
144 [[__gnu__::__always_inline__]]
146 operator*(
const _Vp& __x,
const _Vp& __y)
noexcept
153 [[__gnu__::__always_inline__]]
155 operator/(
const _Vp& __x,
const _Vp& __y)
noexcept
162 [[__gnu__::__always_inline__]]
164 operator%(
const _Vp& __x,
const _Vp& __y)
noexcept
165 requires requires (_Tp __a) { __a % __a; }
172 [[__gnu__::__always_inline__]]
174 operator&(
const _Vp& __x,
const _Vp& __y)
noexcept
175 requires requires (_Tp __a) { __a & __a; }
182 [[__gnu__::__always_inline__]]
184 operator|(
const _Vp& __x,
const _Vp& __y)
noexcept
185 requires requires (_Tp __a) { __a | __a; }
192 [[__gnu__::__always_inline__]]
194 operator^(
const _Vp& __x,
const _Vp& __y)
noexcept
195 requires requires (_Tp __a) { __a ^ __a; }
202 [[__gnu__::__always_inline__]]
204 operator<<(
const _Vp& __x,
const _Vp& __y) _GLIBCXX_SIMD_NOEXCEPT
205 requires requires (_Tp __a) { __a << __a; }
212 [[__gnu__::__always_inline__]]
214 operator<<(
const _Vp& __x, __simd_size_type __y) _GLIBCXX_SIMD_NOEXCEPT
215 requires requires (_Tp __a, __simd_size_type __b) { __a << __b; }
222 [[__gnu__::__always_inline__]]
224 operator>>(
const _Vp& __x,
const _Vp& __y) _GLIBCXX_SIMD_NOEXCEPT
225 requires requires (_Tp __a) { __a >> __a; }
232 [[__gnu__::__always_inline__]]
234 operator>>(
const _Vp& __x, __simd_size_type __y) _GLIBCXX_SIMD_NOEXCEPT
235 requires requires (_Tp __a, __simd_size_type __b) { __a >> __b; }
246 template <
integral _Tp>
247 inline constexpr _Tp __max_shift
248 = (
sizeof(_Tp) <
sizeof(int) ?
sizeof(int) :
sizeof(_Tp)) * __CHAR_BIT__;
250 template <__vectorizable _Tp, __abi_tag _Ap>
251 requires (_Ap::_S_nreg == 1)
252 && (!__complex_like<_Tp>)
253 class basic_vec<_Tp, _Ap>
254 :
public _VecBase<_Tp, _Ap>
256 template <
typename,
typename>
257 friend class basic_vec;
259 template <
size_t,
typename>
260 friend class basic_mask;
262 static constexpr int _S_size = _Ap::_S_size;
264 static constexpr int _S_full_size = __bit_ceil(
unsigned(_S_size));
266 static constexpr bool _S_is_scalar = _S_size == 1;
268 static constexpr bool _S_use_bitmask = _Ap::_S_is_bitmask && !_S_is_scalar;
270 using _DataType =
typename _Ap::template _DataType<_Tp>;
280 static constexpr bool _S_is_partial =
sizeof(_M_data) >
sizeof(_Tp) * _S_size;
282 using __canon_value_type = __canonical_vec_type_t<_Tp>;
285 using value_type = _Tp;
287 using mask_type = _VecBase<_Tp, _Ap>::mask_type;
290 [[__gnu__::__always_inline__]]
291 static constexpr basic_vec
292 _S_init(_DataType __x)
299 [[__gnu__::__always_inline__]]
304 [[__gnu__::__always_inline__]]
305 constexpr const _DataType&
306 _M_get() const noexcept
309 [[__gnu__::__always_inline__]]
310 friend constexpr bool
311 __is_const_known(
const basic_vec& __x)
312 {
return __builtin_constant_p(__x._M_data); }
314 [[__gnu__::__always_inline__]]
316 _M_concat_data([[maybe_unused]]
bool __do_sanitize =
false)
const
318 if constexpr (_S_is_scalar)
319 return __vec_builtin_type<__canon_value_type, 1>{_M_data};
324 template <
int _Size = _S_size,
int _Offset = 0,
typename _A0,
typename _Fp>
325 [[__gnu__::__always_inline__]]
326 static constexpr basic_vec
327 _S_static_permute(
const basic_vec<value_type, _A0>& __x, _Fp&& __idxmap)
329 using _Xp = basic_vec<value_type, _A0>;
331 if constexpr (_S_is_scalar)
333 constexpr __simd_size_type __j = [&]
consteval {
334 if constexpr (__index_permutation_function_sized<_Fp>)
335 return __idxmap(_Offset, _Size);
337 return __idxmap(_Offset);
339 if constexpr (__j == simd::zero_element || __j == simd::uninit_element)
342 static_assert(__j >= 0 && __j < _Xp::_S_size);
343 __r._M_data = __x[__j];
347 auto __idxmap2 = [=](
auto __i)
consteval {
348 if constexpr (int(__i + _Offset) >= _Size)
349 return __simd_size_c<simd::uninit_element>;
350 else if constexpr (__index_permutation_function_sized<_Fp>)
351 return __simd_size_c<__idxmap(__i + _Offset, _Size)>;
353 return __simd_size_c<__idxmap(__i + _Offset)>;
355 constexpr auto __adj_idx = [](
auto __i) {
356 constexpr int __j = __i;
357 if constexpr (__j == simd::zero_element)
358 return __simd_size_c<__bit_ceil(
unsigned(_Xp::_S_size))>;
359 else if constexpr (__j == simd::uninit_element)
360 return __simd_size_c<-1>;
363 static_assert(__j >= 0 && __j < _Xp::_S_size);
364 return __simd_size_c<__j>;
367 constexpr auto [...__is0] = _IotaArray<_S_size>;
368 constexpr bool __needs_zero_element
369 = ((__idxmap2(__simd_size_c<__is0>).value == simd::zero_element) || ...);
370 constexpr auto [...__is_full] = _IotaArray<_S_full_size>;
371 if constexpr (_A0::_S_nreg == 2 && !__needs_zero_element)
373 __r._M_data = __builtin_shufflevector(
374 __x._M_data0._M_data, __x._M_data1._M_data,
375 __adj_idx(__idxmap2(__simd_size_c<__is_full>)).value...);
379 __r._M_data = __builtin_shufflevector(
380 __x._M_concat_data(),
decltype(__x._M_concat_data())(),
381 __adj_idx(__idxmap2(__simd_size_c<__is_full>)).value...);
387 template <
typename _Vp>
388 [[__gnu__::__always_inline__]]
390 _M_chunk() const noexcept
392 constexpr int __n = _S_size / _Vp::_S_size;
393 constexpr int __rem = _S_size % _Vp::_S_size;
394 constexpr auto [...__is] = _IotaArray<__n>;
395 if constexpr (__rem == 0)
396 return array<_Vp, __n> {__extract_simd_at<_Vp>(cw<_Vp::_S_size * __is>, *
this)...};
399 using _Rest = resize_t<__rem, _Vp>;
400 return tuple(__extract_simd_at<_Vp>(cw<_Vp::_S_size * __is>, *
this)...,
401 __extract_simd_at<_Rest>(cw<_Vp::_S_size * __n>, *
this));
405 [[__gnu__::__always_inline__]]
406 static constexpr basic_vec
407 _S_concat(
const basic_vec& __x0)
noexcept
410 template <
typename... _As>
411 requires (
sizeof...(_As) > 1)
412 [[__gnu__::__always_inline__]]
413 static constexpr basic_vec
414 _S_concat(
const basic_vec<value_type, _As>&... __xs)
noexcept
416 static_assert(_S_size == (_As::_S_size + ...));
417 return __extract_simd_at<basic_vec>(cw<0>, __xs...);
444 template <
int _Shift, _ArchTraits _Traits = {}>
445 [[__gnu__::__always_inline__]]
447 _M_elements_shifted_to_front()
const
449 static_assert(_Shift < _S_size && -_Shift < _S_size);
450 if constexpr (_Shift == 0)
453 else if (!__is_const_known(*
this))
455 if constexpr (
sizeof(_M_data) == 16 && _Shift > 0)
456 return reinterpret_cast<_DataType
>(
457 __builtin_ia32_psrldqi128(__vec_bit_cast<long long>(_M_data),
458 _Shift *
sizeof(value_type) * 8));
459 else if constexpr (
sizeof(_M_data) == 16 && _Shift < 0)
460 return reinterpret_cast<_DataType
>(
461 __builtin_ia32_pslldqi128(__vec_bit_cast<long long>(_M_data),
462 -_Shift *
sizeof(value_type) * 8));
463 else if constexpr (
sizeof(_M_data) < 16)
465 auto __x =
reinterpret_cast<__vec_builtin_type_bytes<long long, 16>
>(
466 __vec_zero_pad_to_16(_M_data));
467 if constexpr (_Shift > 0)
468 __x = __builtin_ia32_psrldqi128(__x, _Shift *
sizeof(value_type) * 8);
470 __x = __builtin_ia32_pslldqi128(__x, -_Shift *
sizeof(value_type) * 8);
471 return _VecOps<_DataType>::_S_extract(__vec_bit_cast<__canon_value_type>(__x));
475 return _S_static_permute(*
this, [](
int __i)
consteval {
476 int __off = __i + _Shift;
477 return __off >= _S_size || __off < 0 ? zero_element : __off;
487 template <
typename _Vp, __canon_value_type __
id>
488 [[__gnu__::__always_inline__]]
490 _M_pad_to_T_with_value() const noexcept
492 static_assert(!_Vp::_S_is_partial);
493 static_assert(_Ap::_S_nreg == 1);
494 if constexpr (
sizeof(_Vp) == 32)
496 static_assert(
sizeof(_M_data) == 32);
497 constexpr auto __k = _Vp::mask_type::_S_partial_mask_of_n(_S_size);
498 return __select_impl(__k, _Vp::_S_init(_M_data), __id);
502 static_assert(
sizeof(_Vp) <= 16);
503 static_assert(
sizeof(_M_data) <=
sizeof(_Vp));
504 _Vp __v1 = __vec_zero_pad_to<sizeof(_Vp)>(_M_data);
505 if constexpr (__id == 0 && _S_is_partial)
509 __v1 = __v1.template _M_elements_shifted_to_front<-(_Vp::_S_size - _S_size)>();
510 else if constexpr (_Vp::_S_size - _S_size == 1)
512 __vec_set(__v1._M_data, _Vp::_S_size - 1, __id);
513 else if constexpr (__has_single_bit(
unsigned(_Vp::_S_size - _S_size)))
515 constexpr int __n = _Vp::_S_size - _S_size;
516 using _Ip = __integer_from<__n *
sizeof(__canon_value_type)>;
517 constexpr auto [...__is] = _IotaArray<__n>;
518 constexpr __canon_value_type __idn[__n] = {((void)__is, __id)...};
519 auto __vn = __vec_bit_cast<_Ip>(__v1._M_data);
520 __vec_set(__vn, _Vp::_S_size / __n - 1, __builtin_bit_cast(_Ip, __idn));
521 __v1._M_data =
reinterpret_cast<typename _Vp::_DataType
>(__vn);
523 else if constexpr (__id != 0 && !_S_is_partial)
527 constexpr _Vp __idn([](
int __i) {
528 return __i >= _S_size ? __id : __canon_value_type();
530 __v1._M_data = __vec_or(__v1._M_data, __idn._M_data);
532 else if constexpr (__id != 0 || _S_is_partial)
534 constexpr auto __k = _Vp::mask_type::_S_partial_mask_of_n(_S_size);
535 __v1 = __select_impl(__k, __v1, __id);
541 [[__gnu__::__always_inline__]]
543 _M_reduce_to_half(
auto __binary_op)
const
545 static_assert(__has_single_bit(
unsigned(_S_size)));
546 auto [__a, __b] = chunk<_S_size / 2>(*this);
547 return __binary_op(__a, __b);
550 template <
typename _Rest,
typename _BinaryOp>
551 [[__gnu__::__always_inline__]]
553 _M_reduce_tail(
const _Rest& __rest, _BinaryOp __binary_op)
const
555 if constexpr (_S_is_scalar)
556 return __binary_op(*
this, __rest)._M_data;
557 else if constexpr (_Rest::_S_size == _S_size)
558 return __binary_op(*
this, __rest)._M_reduce(__binary_op);
559 else if constexpr (_Rest::_S_size > _S_size)
561 auto [__a, __b] = __rest.template _M_chunk<basic_vec>();
562 return __binary_op(*
this, __a)._M_reduce_tail(__b, __binary_op);
564 else if constexpr (_Rest::_S_size == 1)
565 return __binary_op(_Rest(_M_reduce(__binary_op)), __rest)[0];
566 else if constexpr (
sizeof(_M_data) <= 16
567 &&
requires { __default_identity_element<__canon_value_type, _BinaryOp>(); })
569 constexpr __canon_value_type __id
570 = __default_identity_element<__canon_value_type, _BinaryOp>();
571 return __binary_op(_M_data, __rest.template _M_pad_to_T_with_value<basic_vec, __id>())
572 ._M_reduce(__binary_op);
575 return _M_reduce_to_half(__binary_op)._M_reduce_tail(__rest, __binary_op);
584 template <
typename _BinaryOp, _ArchTraits _Traits = {}>
585 [[__gnu__::__always_inline__]]
587 _M_reduce(_BinaryOp __binary_op)
const
589 constexpr bool __have_id_elem
590 =
requires { __default_identity_element<__canon_value_type, _BinaryOp>(); };
591 if constexpr (_S_size == 1)
592 return operator[](0);
593 else if constexpr (_Traits.template _M_eval_as_f32<value_type>()
594 && (is_same_v<_BinaryOp, plus<>>
595 || is_same_v<_BinaryOp, multiplies<>>))
596 return value_type(rebind_t<float, basic_vec>(*this)._M_reduce(__binary_op));
598 else if constexpr (is_integral_v<value_type> &&
sizeof(value_type) == 1
599 && is_same_v<
decltype(__binary_op), multiplies<>>)
614 if constexpr (!_S_is_partial)
616 using _V16 = resize_t<_S_size / 2, rebind_t<unsigned short, basic_vec>>;
617 auto __a = __builtin_bit_cast(_V16, *
this);
618 return __binary_op(__a, __a >> 8)._M_reduce(__binary_op);
622 using _V16 = rebind_t<unsigned short, basic_vec>;
623 return _V16(*this)._M_reduce(__binary_op);
627 else if constexpr (__has_single_bit(
unsigned(_S_size)))
629 if constexpr (
sizeof(_M_data) > 16)
630 return _M_reduce_to_half(__binary_op)._M_reduce(__binary_op);
631 else if constexpr (_S_size == 2)
632 return _M_reduce_to_half(__binary_op)[0];
635 static_assert(_S_size <= 16);
638 if constexpr (
sizeof(_M_data) <= 16 && is_integral_v<value_type>)
640 if constexpr (_S_size > 8)
641 __x = __binary_op(__x, __x.template _M_elements_shifted_to_front<8>());
642 if constexpr (_S_size > 4)
643 __x = __binary_op(__x, __x.template _M_elements_shifted_to_front<4>());
644 if constexpr (_S_size > 2)
645 __x = __binary_op(__x, __x.template _M_elements_shifted_to_front<2>());
648 return __binary_op(__x, __x.template _M_elements_shifted_to_front<1>())[0];
651 if constexpr (_S_size > 8)
652 __x = __binary_op(__x, _S_static_permute(__x, _SwapNeighbors<8>()));
653 if constexpr (_S_size > 4)
654 __x = __binary_op(__x, _S_static_permute(__x, _SwapNeighbors<4>()));
657 if constexpr (is_integral_v<value_type> &&
sizeof(value_type) <= 1)
658 return value_type(resize_t<4, rebind_t<int, basic_vec>>(chunk<4>(__x)[0])
659 ._M_reduce(__binary_op));
661 if constexpr (_S_size > 2)
662 __x = __binary_op(__x, _S_static_permute(__x, _SwapNeighbors<2>()));
663 if constexpr (is_integral_v<value_type> &&
sizeof(value_type) == 2)
664 return __binary_op(__x, _S_static_permute(__x, _SwapNeighbors<1>()))[0];
666 return __binary_op(vec<value_type, 1>(__x[0]), vec<value_type, 1>(__x[1]))[0];
669 else if constexpr (
sizeof(_M_data) == 32)
671 const auto [__lo, __hi] = chunk<__bit_floor(unsigned(_S_size))>(*
this);
672 return __lo._M_reduce_tail(__hi, __binary_op);
674 else if constexpr (
sizeof(_M_data) == 64)
678 auto __chunked = chunk<__bit_floor(
unsigned(_S_size)) / 2>(*this);
679 using _Cp =
decltype(__chunked);
680 if constexpr (tuple_size_v<_Cp> == 4)
682 const auto& [__a, __b, __c, __rest] = __chunked;
683 constexpr bool __amd_cpu = _Traits._M_have_sse4a();
684 if constexpr (__have_id_elem && __rest._S_size > 1 && __amd_cpu)
690 const auto& [__a, __rest] = chunk<__bit_floor(unsigned(_S_size))>(*
this);
691 using _Vp = remove_cvref_t<
decltype(__a)>;
692 constexpr __canon_value_type __id
693 = __default_identity_element<__canon_value_type, _BinaryOp>();
694 const _Vp __b = __rest.template _M_pad_to_T_with_value<_Vp, __id>();
695 return __binary_op(__a, __b)._M_reduce(__binary_op);
697 else if constexpr (__have_id_elem && __rest._S_size > 1)
703 using _Vp = remove_cvref_t<
decltype(__a)>;
704 constexpr __canon_value_type __id
705 = __default_identity_element<__canon_value_type, _BinaryOp>();
706 const _Vp __d = __rest.template _M_pad_to_T_with_value<_Vp, __id>();
707 return __binary_op(__binary_op(__a, __b), __binary_op(__c, __d))
708 ._M_reduce(__binary_op);
711 return __binary_op(__binary_op(__a, __b), __c)
712 ._M_reduce_tail(__rest, __binary_op);
714 else if constexpr (tuple_size_v<_Cp> == 3)
716 const auto& [__a, __b, __rest] = __chunked;
717 return __binary_op(__a, __b)._M_reduce_tail(__rest, __binary_op);
720 static_assert(
false);
722 else if constexpr (__have_id_elem)
724 constexpr __canon_value_type __id
725 = __default_identity_element<__canon_value_type, _BinaryOp>();
726 using _Vp = resize_t<__bit_ceil(
unsigned(_S_size)), basic_vec>;
727 return _M_pad_to_T_with_value<_Vp, __id>()._M_reduce(__binary_op);
731 const auto& [__a, __rest] = chunk<__bit_floor(unsigned(_S_size))>(*
this);
732 return __a._M_reduce_tail(__rest, __binary_op);
741 template <_OptTraits _Traits = {}>
742 [[__gnu__::__always_inline__]]
744 _M_isnan() const requires is_floating_point_v<value_type>
746 if constexpr (_Traits._M_finite_math_only())
747 return mask_type(
false);
748 else if constexpr (_S_is_scalar)
749 return mask_type(std::isnan(_M_data));
750 else if constexpr (_S_use_bitmask)
751 return _M_isunordered(*
this);
752 else if constexpr (!_Traits._M_support_snan())
753 return !(*
this == *
this);
754 else if (__is_const_known(_M_data))
755 return mask_type([&](
int __i) {
return std::isnan(_M_data[__i]); });
759 using _Ip = __integer_from<
sizeof(value_type)>;
761 < __builtin_bit_cast(rebind_t<_Ip, basic_vec>, _M_fabs());
765 template <_TargetTraits _Traits = {}>
766 [[__gnu__::__always_inline__]]
768 _M_isinf() const requires is_floating_point_v<value_type>
770 if constexpr (_Traits._M_finite_math_only())
771 return mask_type(
false);
772 else if constexpr (_S_is_scalar)
773 return mask_type(std::isinf(_M_data));
774 else if (__is_const_known(_M_data))
775 return mask_type([&](
int __i) {
return std::isinf(_M_data[__i]); });
777 else if constexpr (_S_use_bitmask)
778 return mask_type::_S_init(__x86_bitmask_isinf(_M_data));
779 else if constexpr (_Traits._M_have_avx512dq())
780 return __x86_bit_to_vecmask<typename mask_type::_DataType>(
781 __x86_bitmask_isinf(_M_data));
785 using _Ip = __integer_from<
sizeof(value_type)>;
786 return __vec_bit_cast<_Ip>(_M_fabs()._M_data)
791 [[__gnu__::__always_inline__]]
793 _M_abs() const requires signed_integral<value_type>
794 {
return _M_data < 0 ? -_M_data : _M_data; }
796 [[__gnu__::__always_inline__]]
798 _M_fabs() const requires floating_point<value_type>
800 if constexpr (_S_is_scalar)
803 return __vec_and(__vec_not(_S_signmask<_DataType>), _M_data);
806 template <_TargetTraits _Traits = {}>
807 [[__gnu__::__always_inline__]]
809 _M_isunordered(basic_vec __y)
const requires is_floating_point_v<value_type>
811 if constexpr (_Traits._M_finite_math_only())
812 return mask_type(
false);
813 else if constexpr (_S_is_scalar)
814 return mask_type(std::isunordered(_M_data, __y._M_data));
816 else if constexpr (_S_use_bitmask)
817 return _M_bitmask_cmp<_X86Cmp::_Unord>(__y._M_data);
820 return mask_type([&](
int __i) {
821 return std::isunordered(_M_data[__i], __y._M_data[__i]);
832 template <
typename _Up, _ArchTraits _Traits = {}>
833 static inline basic_vec
834 _S_partial_load(
const _Up* __mem,
size_t __n)
836 if constexpr (_S_is_scalar)
837 return __n == 0 ? basic_vec() : basic_vec(
static_cast<value_type
>(*__mem));
838 else if (__is_const_known_equal_to(__n >=
size_t(_S_size),
true))
839 return basic_vec(_LoadCtorTag(), __mem);
840 else if constexpr (!__converts_trivially<_Up, value_type>)
841 return static_cast<basic_vec
>(rebind_t<_Up, basic_vec>::_S_partial_load(__mem, __n));
845 if constexpr (_Traits._M_have_avx512f()
846 || (_Traits._M_have_avx() &&
sizeof(_Up) >= 4))
848 const auto __k = __n < _S_size ? mask_type::_S_partial_mask_of_n(
int(__n))
850 return _S_masked_load(__mem, mask_type::_S_partial_mask_of_n(
int(__n)));
853 if (__n >=
size_t(_S_size)) [[unlikely]]
854 return basic_vec(_LoadCtorTag(), __mem);
857 else if (__is_const_known_equal_to(
859 return __select_impl(mask_type::_S_partial_mask_of_n(
int(__n)),
860 basic_vec(_LoadCtorTag(), __mem), basic_vec());
862 else if constexpr (_S_size > 4)
864 alignas(_DataType)
byte __dst[
sizeof(_DataType)] = {};
865 const byte* __src =
reinterpret_cast<const byte*
>(__mem);
866 __memcpy_chunks<sizeof(_Up), sizeof(_DataType)>(__dst, __src, __n);
867 return __builtin_bit_cast(_DataType, __dst);
869 else if (__n == 0) [[unlikely]]
871 else if constexpr (_S_size == 2)
872 return _DataType {
static_cast<value_type
>(__mem[0]), 0};
875 constexpr auto [...__is] = _IotaArray<_S_size - 2>;
877 static_cast<value_type
>(__mem[0]),
878 static_cast<value_type
>(__is + 1 < __n ? __mem[__is + 1] : 0)...
896 template <
typename _Up, _ArchTraits _Traits = {}>
897 static inline basic_vec
898 _S_masked_load(
const _Up* __mem, mask_type __k)
900 if constexpr (_S_size == 1)
901 return __k[0] ?
static_cast<value_type
>(__mem[0]) : value_type();
903 else if constexpr (_Traits._M_have_avx512f())
904 return __x86_masked_load<_DataType>(__mem, __k._M_data);
905 else if constexpr (_Traits._M_have_avx() && (
sizeof(_Up) == 4 ||
sizeof(_Up) == 8))
907 if constexpr (__converts_trivially<_Up, value_type>)
908 return __x86_masked_load<_DataType>(__mem, __k._M_data);
911 using _UV = rebind_t<_Up, basic_vec>;
912 return basic_vec(_UV::_S_masked_load(__mem,
typename _UV::mask_type(__k)));
916 else if (__k._M_none_of()) [[unlikely]]
918 else if constexpr (_S_is_scalar)
919 return basic_vec(
static_cast<value_type
>(*__mem));
923 _Bitmask<_S_size < 32 ? 32 : _S_size> __bits = __k._M_to_uint();
924 [[assume(__bits != 0)]];
925 if constexpr (__converts_trivially<_Up, value_type>)
928 __bit_foreach(__bits, [&] [[__gnu__::__always_inline__]] (
int __i) {
929 __r[__i] = __mem[__i];
935 using _UV = rebind_t<_Up, basic_vec>;
936 alignas(_UV) _Up __tmp[
sizeof(_UV) /
sizeof(_Up)] = {};
937 __bit_foreach(__bits, [&] [[__gnu__::__always_inline__]] (
int __i) {
938 __tmp[__i] = __mem[__i];
940 return basic_vec(__builtin_bit_cast(_UV, __tmp));
945 template <
typename _Up>
946 [[__gnu__::__always_inline__]]
948 _M_store(_Up* __mem)
const
950 if constexpr (__converts_trivially<value_type, _Up>)
951 __builtin_memcpy(__mem, &_M_data,
sizeof(_Up) * _S_size);
953 rebind_t<_Up, basic_vec>(*this)._M_store(__mem);
965 template <
typename _Up, _ArchTraits _Traits = {}>
967 _S_partial_store(
const basic_vec __v, _Up* __mem,
size_t __n)
969 if (__is_const_known_equal_to(__n >= _S_size,
true))
972 else if constexpr (_Traits._M_have_avx512f() && !_S_is_scalar)
974 const auto __k = __n < _S_size ? mask_type::_S_partial_mask_of_n(
int(__n))
976 return _S_masked_store(__v, __mem, __k);
979 else if (__n >= _S_size) [[unlikely]]
981 else if (__n == 0) [[unlikely]]
983 else if constexpr (__converts_trivially<value_type, _Up>)
985 byte* __dst =
reinterpret_cast<byte*
>(__mem);
986 const byte* __src =
reinterpret_cast<const byte*
>(&__v._M_data);
987 __memcpy_chunks<sizeof(_Up), sizeof(_M_data)>(__dst, __src, __n);
991 using _UV = rebind_t<_Up, basic_vec>;
992 _UV::_S_partial_store(_UV(__v), __mem, __n);
1009 template <
typename _Up, _ArchTraits _Traits = {}>
1012 _S_masked_store(
const basic_vec __v, _Up* __mem,
const mask_type __k)
1015 if constexpr (_Traits._M_have_avx512f())
1017 __x86_masked_store(__v._M_data, __mem, __k._M_data);
1020 else if constexpr (_Traits._M_have_avx() && (
sizeof(_Up) == 4 ||
sizeof(_Up) == 8))
1022 if constexpr (__converts_trivially<value_type, _Up>)
1023 __x86_masked_store(__v._M_data, __mem, __k._M_data);
1026 using _UV = rebind_t<_Up, basic_vec>;
1027 _UV::_S_masked_store(_UV(__v), __mem,
typename _UV::mask_type(__k));
1032 if (__k._M_none_of()) [[unlikely]]
1034 else if constexpr (_S_is_scalar)
1035 __mem[0] = __v._M_data;
1039 _Bitmask<_S_size < 32 ? 32 : _S_size> __bits = __k._M_to_uint();
1040 [[assume(__bits != 0)]];
1041 if constexpr (__converts_trivially<value_type, _Up>)
1043 __bit_foreach(__bits, [&] [[__gnu__::__always_inline__]] (
int __i) {
1044 __mem[__i] = __v[__i];
1049 const rebind_t<_Up, basic_vec> __cvted(__v);
1050 __bit_foreach(__bits, [&] [[__gnu__::__always_inline__]] (
int __i) {
1051 __mem[__i] = __cvted[__i];
1058 basic_vec() =
default;
1061 using _NativeVecType =
decltype([] {
1062 if constexpr (_S_is_scalar)
1063 return __vec_builtin_type<__canon_value_type, 1>();
1080 basic_vec(_NativeVecType __x)
1081 : _M_data([&] [[__gnu__::__always_inline__]] {
1082 if constexpr (_S_is_scalar)
1099 operator _NativeVecType()
const
1101 if constexpr (_S_is_scalar)
1102 return _NativeVecType{_M_data};
1111 template <__vec_builtin _IV>
1112 requires same_as<__x86_intel_intrin_value_type<value_type>, __vec_value_type<_IV>>
1113 && (
sizeof(_IV) ==
sizeof(_DataType) &&
sizeof(_IV) >= 16
1114 && !is_same_v<_IV, _DataType>)
1117 : _M_data(reinterpret_cast<_DataType>(__x))
1123 template <__vec_builtin _IV>
1124 requires same_as<__x86_intel_intrin_value_type<value_type>, __vec_value_type<_IV>>
1125 && (
sizeof(_IV) ==
sizeof(_DataType) &&
sizeof(_IV) >= 16
1126 && !is_same_v<_IV, _DataType>)
1128 operator _IV()
const
1129 {
return reinterpret_cast<_IV
>(_M_data); }
1144 template <__broadcast_constructible<value_type> _Up>
1145 [[__gnu__::__always_inline__]]
1147 basic_vec(_Up&& __x) noexcept
1148 : _M_data(_DataType() == _DataType() ?
static_cast<value_type
>(__x) : value_type())
1152 template <
typename _Up,
typename _UAbi, _TargetTraits _Traits = {}>
1153 requires (_S_size == _UAbi::_S_size)
1154 && __explicitly_convertible_to<_Up, value_type>
1155 [[__gnu__::__always_inline__]]
1157 explicit(!__value_preserving_convertible_to<_Up, value_type>
1158 || __higher_rank_than<_Up, value_type>)
1159 basic_vec(
const basic_vec<_Up, _UAbi>& __x) noexcept
1160 : _M_data([&] [[__gnu__::__always_inline__]] {
1161 if constexpr (_S_is_scalar)
1162 return static_cast<value_type
>(__x[0]);
1163 else if constexpr (_UAbi::_S_nreg >= 2)
1167 return _S_concat(resize_t<__x._N0, basic_vec>(__x._M_data0),
1168 resize_t<__x._N1, basic_vec>(__x._M_data1))._M_data;
1170 return __vec_cast<_DataType>(__x._M_concat_data());
1174 using _VecBase<_Tp, _Ap>::_VecBase;
1177 template <__simd_generator_invokable<value_type, _S_size> _Fp>
1178 [[__gnu__::__always_inline__]]
1180 basic_vec(_Fp&& __gen)
1181 : _M_data([&] [[__gnu__::__always_inline__]] {
1182 constexpr auto [...__is] = _IotaArray<_S_size>;
1183 return _DataType{
static_cast<value_type
>(__gen(__simd_size_c<__is>))...};
1188 template <
typename _Up>
1189 [[__gnu__::__always_inline__]]
1191 basic_vec(_LoadCtorTag,
const _Up* __ptr)
1194 if constexpr (_S_is_scalar)
1195 _M_data =
static_cast<value_type
>(__ptr[0]);
1198 constexpr auto [...__is] = _IotaArray<_S_size>;
1199 _M_data = _DataType{
static_cast<value_type
>(__ptr[__is])...};
1203 if constexpr (__converts_trivially<_Up, value_type>)
1205 __builtin_memcpy(&_M_data, __ptr,
sizeof(value_type) * _S_size);
1208 __vec_builtin_type<_Up, _S_full_size> __tmp = {};
1209 __builtin_memcpy(&__tmp, __ptr,
sizeof(_Up) * _S_size);
1210 _M_data = __vec_cast<_DataType>(__tmp);
1215 template <ranges::contiguous_range _Rg,
typename... _Flags>
1216 requires ranges::__static_sized_range<_Rg>
1217 && __vectorizable<ranges::range_value_t<_Rg>>
1218 && __explicitly_convertible_to<ranges::range_value_t<_Rg>, value_type>
1219 [[__gnu__::__always_inline__]]
1221 basic_vec(_Rg&& __range, flags<_Flags...> __flags = {})
1222 requires (ranges::size(__range) == _S_size)
1223 : basic_vec(_LoadCtorTag(), __flags.template _S_adjust_pointer<basic_vec>(
1224 ranges::
data(__range)))
1226 static_assert(__loadstore_convertible_to<ranges::range_value_t<_Rg>, value_type,
1236 [[__gnu__::__always_inline__]]
1237 constexpr value_type
1238 operator[](__simd_size_type __i)
const
1240 __glibcxx_simd_precondition(__i >= 0 && __i < _S_size,
"subscript is out of bounds");
1241 if constexpr (_S_is_scalar)
1244 return _M_data[__i];
1250 [[__gnu__::__always_inline__]]
1251 constexpr basic_vec&
1252 operator++() noexcept requires requires(value_type __a) { ++__a; }
1253 {
return *
this += value_type(1); }
1255 [[__gnu__::__always_inline__]]
1257 operator++(
int)
noexcept requires requires(value_type __a) { __a++; }
1259 basic_vec __r = *
this;
1260 *
this += value_type(1);
1264 [[__gnu__::__always_inline__]]
1265 constexpr basic_vec&
1266 operator--() noexcept requires requires(value_type __a) { --__a; }
1267 {
return *
this -= value_type(1); }
1269 [[__gnu__::__always_inline__]]
1271 operator--(
int)
noexcept requires requires(value_type __a) { __a--; }
1273 basic_vec __r = *
this;
1274 *
this -= value_type(1);
1278 [[__gnu__::__always_inline__]]
1280 operator!() const noexcept requires requires(value_type __a) { !__a; }
1281 {
return *
this == value_type(); }
1288 [[__gnu__::__always_inline__]]
1290 operator+() const noexcept requires requires(value_type __a) { +__a; }
1298 [[__gnu__::__always_inline__]]
1300 operator-() const noexcept requires requires(value_type __a) { -__a; }
1301 {
return _S_init(-_M_data); }
1308 [[__gnu__::__always_inline__]]
1310 operator~() const noexcept requires requires(value_type __a) { ~__a; }
1311 {
return _S_init(~_M_data); }
1319 [[__gnu__::__always_inline__]]
1320 friend constexpr basic_vec&
1321 operator&=(basic_vec& __x,
const basic_vec& __y)
noexcept
1322 requires requires(value_type __a) { __a & __a; }
1324 __x._M_data &= __y._M_data;
1333 [[__gnu__::__always_inline__]]
1334 friend constexpr basic_vec&
1335 operator|=(basic_vec& __x,
const basic_vec& __y)
noexcept
1336 requires requires(value_type __a) { __a | __a; }
1338 __x._M_data |= __y._M_data;
1347 [[__gnu__::__always_inline__]]
1348 friend constexpr basic_vec&
1349 operator^=(basic_vec& __x,
const basic_vec& __y)
noexcept
1350 requires requires(value_type __a) { __a ^ __a; }
1352 __x._M_data ^= __y._M_data;
1365 [[__gnu__::__always_inline__]]
1366 friend constexpr basic_vec&
1367 operator+=(basic_vec& __x,
const basic_vec& __y)
noexcept
1368 requires requires(value_type __a) { __a + __a; }
1370 if constexpr (_S_is_partial && is_integral_v<value_type> && is_signed_v<value_type>)
1385 using _UV =
typename _Ap::template _DataType<make_unsigned_t<value_type>>;
1386 const _DataType __result
1387 =
reinterpret_cast<_DataType
>(
reinterpret_cast<_UV
>(__x._M_data)
1388 +
reinterpret_cast<_UV
>(__y._M_data));
1389 const auto __positive = __y > value_type();
1390 const auto __overflow = __positive != (__result > __x);
1391 if (__overflow._M_any_of())
1392 __builtin_unreachable();
1393 __x._M_data = __result;
1395 else if constexpr (_TargetTraits()._M_eval_as_f32<value_type>())
1396 __x = basic_vec(rebind_t<float, basic_vec>(__x) + __y);
1398 __x._M_data += __y._M_data;
1404 [[__gnu__::__always_inline__]]
1405 friend constexpr basic_vec&
1406 operator-=(basic_vec& __x,
const basic_vec& __y)
noexcept
1407 requires requires(value_type __a) { __a - __a; }
1409 if constexpr (_S_is_partial && is_integral_v<value_type> && is_signed_v<value_type>)
1411 using _UV =
typename _Ap::template _DataType<make_unsigned_t<value_type>>;
1412 const _DataType __result
1413 =
reinterpret_cast<_DataType
>(
reinterpret_cast<_UV
>(__x._M_data)
1414 -
reinterpret_cast<_UV
>(__y._M_data));
1415 const auto __positive = __y > value_type();
1416 const auto __overflow = __positive != (__result < __x);
1417 if (__overflow._M_any_of())
1418 __builtin_unreachable();
1419 __x._M_data = __result;
1421 else if constexpr (_TargetTraits()._M_eval_as_f32<value_type>())
1422 __x = basic_vec(rebind_t<float, basic_vec>(__x) - __y);
1424 __x._M_data -= __y._M_data;
1430 [[__gnu__::__always_inline__]]
1431 friend constexpr basic_vec&
1432 operator*=(basic_vec& __x,
const basic_vec& __y)
noexcept
1433 requires requires(value_type __a) { __a * __a; }
1435 if constexpr (_S_is_partial && is_integral_v<value_type> && is_signed_v<value_type>)
1437 for (
int __i = 0; __i < _S_size; ++__i)
1439 if (__builtin_mul_overflow_p(__x._M_data[__i], __y._M_data[__i], value_type()))
1440 __builtin_unreachable();
1442 using _UV =
typename _Ap::template _DataType<make_unsigned_t<value_type>>;
1443 __x._M_data =
reinterpret_cast<_DataType
>(
reinterpret_cast<_UV
>(__x._M_data)
1444 *
reinterpret_cast<_UV
>(__y._M_data));
1450 else if constexpr (_S_is_scalar && is_unsigned_v<value_type>
1451 && is_signed_v<
decltype(value_type() * value_type())>)
1452 __x._M_data =
unsigned(__x._M_data) * unsigned(__y._M_data);
1454 else if constexpr (_TargetTraits()._M_eval_as_f32<value_type>())
1455 __x = basic_vec(rebind_t<float, basic_vec>(__x) * __y);
1458 __x._M_data *= __y._M_data;
1462 template <_TargetTraits _Traits = {}>
1463 [[__gnu__::__always_inline__]]
1464 friend constexpr basic_vec&
1465 operator/=(basic_vec& __x,
const basic_vec& __y)
noexcept
1466 requires requires(value_type __a) { __a / __a; }
1468 const basic_vec __result([&](
int __i) -> value_type {
return __x[__i] / __y[__i]; });
1469 if (__is_const_known(__result))
1471 return __x = __result;
1479 if constexpr (is_integral_v<value_type> && _S_size > 2
1480 && __value_preserving_convertible_to<value_type, double>)
1484 if (!__is_const_known(__y))
1487 if constexpr (_Traits._M_have_avx512fp16()
1488 && __value_preserving_convertible_to<value_type, _Float16>)
1489 return __x = basic_vec(rebind_t<_Float16, basic_vec>(__x) / __y);
1490 else if constexpr (__value_preserving_convertible_to<value_type, float>)
1491 return __x = basic_vec(rebind_t<float, basic_vec>(__x) / __y);
1493 return __x = basic_vec(rebind_t<double, basic_vec>(__x) / __y);
1497 if constexpr (_Traits._M_eval_as_f32<value_type>())
1498 return __x = basic_vec(rebind_t<float, basic_vec>(__x) / __y);
1500 basic_vec __y1 = __y;
1501 if constexpr (_S_is_partial)
1503 if constexpr (is_integral_v<value_type>)
1507 for (
int __i = 0; __i < _S_size; ++__i)
1508 __x._M_data[__i] /= __y._M_data[__i];
1512 __y1 = __select_impl(mask_type::_S_init(mask_type::_S_implicit_mask),
1513 __y, basic_vec(value_type(1)));
1515 __x._M_data /= __y1._M_data;
1519 [[__gnu__::__always_inline__]]
1520 friend constexpr basic_vec&
1521 operator%=(basic_vec& __x,
const basic_vec& __y)
noexcept
1522 requires requires(value_type __a) { __a % __a; }
1524 static_assert(is_integral_v<value_type>);
1525 if constexpr (_S_is_partial)
1527 const basic_vec __y1 = __select_impl(mask_type::_S_init(mask_type::_S_implicit_mask),
1528 __y, basic_vec(value_type(1)));
1529 if (__is_const_known(__y1))
1530 __x._M_data %= __y1._M_data;
1535 for (
int __i = 0; __i < _S_size; ++__i)
1536 __x._M_data[__i] %= __y._M_data[__i];
1540 __x._M_data %= __y._M_data;
1544 [[__gnu__::__always_inline__]]
1545 friend constexpr basic_vec&
1546 operator<<=(basic_vec& __x,
const basic_vec& __y) _GLIBCXX_SIMD_NOEXCEPT
1547 requires requires(value_type __a) { __a << __a; }
1549 __glibcxx_simd_precondition(is_unsigned_v<value_type> || all_of(__y >= value_type()),
1550 "negative shift is undefined behavior");
1551 __glibcxx_simd_precondition(all_of(__y < __max_shift<value_type>),
1552 "too large shift invokes undefined behavior");
1553 __x._M_data <<= __y._M_data;
1557 [[__gnu__::__always_inline__]]
1558 friend constexpr basic_vec&
1559 operator>>=(basic_vec& __x,
const basic_vec& __y) _GLIBCXX_SIMD_NOEXCEPT
1560 requires requires(value_type __a) { __a >> __a; }
1562 __glibcxx_simd_precondition(is_unsigned_v<value_type> || all_of(__y >= value_type()),
1563 "negative shift is undefined behavior");
1564 __glibcxx_simd_precondition(all_of(__y < __max_shift<value_type>),
1565 "too large shift invokes undefined behavior");
1566 __x._M_data >>= __y._M_data;
1570 [[__gnu__::__always_inline__]]
1571 friend constexpr basic_vec&
1572 operator<<=(basic_vec& __x, __simd_size_type __y) _GLIBCXX_SIMD_NOEXCEPT
1573 requires requires(value_type __a, __simd_size_type __b) { __a << __b; }
1575 __glibcxx_simd_precondition(__y >= 0,
"negative shift is undefined behavior");
1576 __glibcxx_simd_precondition(__y <
int(__max_shift<value_type>),
1577 "too large shift invokes undefined behavior");
1578 __x._M_data <<= __y;
1582 [[__gnu__::__always_inline__]]
1583 friend constexpr basic_vec&
1584 operator>>=(basic_vec& __x, __simd_size_type __y) _GLIBCXX_SIMD_NOEXCEPT
1585 requires requires(value_type __a, __simd_size_type __b) { __a >> __b; }
1587 __glibcxx_simd_precondition(__y >= 0,
"negative shift is undefined behavior");
1588 __glibcxx_simd_precondition(__y <
int(__max_shift<value_type>),
1589 "too large shift invokes undefined behavior");
1590 __x._M_data >>= __y;
1596 template <_X86Cmp _Cmp>
1597 [[__gnu__::__always_inline__]]
1599 _M_bitmask_cmp(_DataType __y)
const
1601 static_assert(_S_use_bitmask);
1602 if (__is_const_known(_M_data, __y))
1604 constexpr auto [...__is] = _IotaArray<_S_size>;
1605 constexpr auto __cmp_op = [] [[__gnu__::__always_inline__]]
1606 (value_type __a, value_type __b) {
1607 if constexpr (_Cmp == _X86Cmp::_Eq)
1609 else if constexpr (_Cmp == _X86Cmp::_Lt)
1611 else if constexpr (_Cmp == _X86Cmp::_Le)
1613 else if constexpr (_Cmp == _X86Cmp::_Unord)
1614 return std::isunordered(__a, __b);
1615 else if constexpr (_Cmp == _X86Cmp::_Neq)
1617 else if constexpr (_Cmp == _X86Cmp::_Nlt)
1618 return !(__a < __b);
1619 else if constexpr (_Cmp == _X86Cmp::_Nle)
1620 return !(__a <= __b);
1622 static_assert(
false);
1624 const _Bitmask<_S_size> __bits
1625 = ((__cmp_op(__vec_get(_M_data, __is), __vec_get(__y, __is))
1626 ? (1ULL << __is) : 0) | ...);
1627 return mask_type::_S_init(__bits);
1630 return mask_type::_S_init(__x86_bitmask_cmp<_Cmp>(_M_data, __y));
1634 [[__gnu__::__always_inline__]]
1635 friend constexpr mask_type
1636 operator==(
const basic_vec& __x,
const basic_vec& __y)
noexcept
1639 if constexpr (_S_use_bitmask)
1640 return __x._M_bitmask_cmp<_X86Cmp::_Eq>(__y._M_data);
1643 return mask_type::_S_init(__x._M_data == __y._M_data);
1646 [[__gnu__::__always_inline__]]
1647 friend constexpr mask_type
1648 operator!=(
const basic_vec& __x,
const basic_vec& __y)
noexcept
1651 if constexpr (_S_use_bitmask)
1652 return __x._M_bitmask_cmp<_X86Cmp::_Neq>(__y._M_data);
1655 return mask_type::_S_init(__x._M_data != __y._M_data);
1658 [[__gnu__::__always_inline__]]
1659 friend constexpr mask_type
1660 operator<(
const basic_vec& __x,
const basic_vec& __y)
noexcept
1663 if constexpr (_S_use_bitmask)
1664 return __x._M_bitmask_cmp<_X86Cmp::_Lt>(__y._M_data);
1667 return mask_type::_S_init(__x._M_data < __y._M_data);
1670 [[__gnu__::__always_inline__]]
1671 friend constexpr mask_type
1672 operator<=(
const basic_vec& __x,
const basic_vec& __y)
noexcept
1675 if constexpr (_S_use_bitmask)
1676 return __x._M_bitmask_cmp<_X86Cmp::_Le>(__y._M_data);
1679 return mask_type::_S_init(__x._M_data <= __y._M_data);
1682 [[__gnu__::__always_inline__]]
1683 friend constexpr mask_type
1684 operator>(
const basic_vec& __x,
const basic_vec& __y)
noexcept
1685 {
return __y < __x; }
1687 [[__gnu__::__always_inline__]]
1688 friend constexpr mask_type
1689 operator>=(
const basic_vec& __x,
const basic_vec& __y)
noexcept
1690 {
return __y <= __x; }
1693 template <_TargetTraits _Traits = {}>
1694 [[__gnu__::__always_inline__]]
1695 friend constexpr basic_vec
1696 __select_impl(
const mask_type& __k,
const basic_vec& __t,
const basic_vec& __f)
noexcept
1698 if constexpr (_S_size == 1)
1699 return __k[0] ? __t : __f;
1700 else if constexpr (_S_use_bitmask)
1703 if (__is_const_known(__k, __t, __f))
1704 return basic_vec([&](
int __i) {
return __k[__i] ? __t[__i] : __f[__i]; });
1706 return __x86_bitmask_blend(__k._M_data, __t._M_data, __f._M_data);
1708 static_assert(
false,
"TODO");
1713 return __k._M_data ? __t._M_data : __f._M_data;
1717 constexpr bool __uses_simd_register =
sizeof(_M_data) >= 8;
1718 using _VO = _VecOps<_DataType>;
1719 if (_VO::_S_is_const_known_equal_to(__f._M_data, 0))
1721 if (is_integral_v<value_type> && __uses_simd_register
1722 && _VO::_S_is_const_known_equal_to(__t._M_data, 1))
1727 return basic_vec((-__k)._M_abs());
1729 return __vec_and(
reinterpret_cast<_DataType
>(__k._M_data), __t._M_data);
1731 else if (_VecOps<_DataType>::_S_is_const_known_equal_to(__t._M_data, 0))
1733 if (is_integral_v<value_type> && __uses_simd_register
1734 && _VO::_S_is_const_known_equal_to(__f._M_data, 1))
1735 return value_type(1) + basic_vec(-__k);
1737 return __vec_and(
reinterpret_cast<_DataType
>(__vec_not(__k._M_data)), __f._M_data);
1746 return __k._M_data < 0 ? __t._M_data : __f._M_data;
1748 return __k._M_data ? __t._M_data : __f._M_data;
1754 template <__vectorizable _Tp, __abi_tag _Ap>
1755 requires (_Ap::_S_nreg > 1)
1756 && (!__complex_like<_Tp>)
1757 class basic_vec<_Tp, _Ap>
1758 :
public _VecBase<_Tp, _Ap>
1760 template <
typename,
typename>
1761 friend class basic_vec;
1763 template <
size_t,
typename>
1764 friend class basic_mask;
1766 static constexpr int _S_size = _Ap::_S_size;
1768 static constexpr int _N0 = __bit_ceil(
unsigned(_S_size)) / 2;
1770 static constexpr int _N1 = _S_size - _N0;
1772 using _DataType0 = __similar_vec<_Tp, _N0, _Ap>;
1775 static_assert(_N0 *
sizeof(_Tp) ==
sizeof(_DataType0));
1777 using _DataType1 = __similar_vec<_Tp, _N1, _Ap>;
1779 static_assert(_DataType0::abi_type::_S_nreg + _DataType1::abi_type::_S_nreg == _Ap::_S_nreg);
1781 static constexpr bool _S_is_scalar = _DataType0::_S_is_scalar;
1783 _DataType0 _M_data0;
1785 _DataType1 _M_data1;
1787 static constexpr bool _S_use_bitmask = _DataType0::_S_use_bitmask;
1789 static constexpr bool _S_is_partial = _DataType1::_S_is_partial;
1792 using value_type = _Tp;
1794 using mask_type = _VecBase<_Tp, _Ap>::mask_type;
1796 [[__gnu__::__always_inline__]]
1797 static constexpr basic_vec
1798 _S_init(
const _DataType0& __x,
const _DataType1& __y)
1806 [[__gnu__::__always_inline__]]
1807 constexpr _DataType0&
1808 _M_get_low() noexcept
1809 {
return _M_data0; }
1811 [[__gnu__::__always_inline__]]
1812 constexpr const _DataType0&
1813 _M_get_low() const noexcept
1814 {
return _M_data0; }
1816 [[__gnu__::__always_inline__]]
1817 constexpr _DataType1&
1818 _M_get_high() noexcept
1819 {
return _M_data1; }
1821 [[__gnu__::__always_inline__]]
1822 constexpr const _DataType1&
1823 _M_get_high() const noexcept
1824 {
return _M_data1; }
1826 [[__gnu__::__always_inline__]]
1827 friend constexpr bool
1828 __is_const_known(
const basic_vec& __x)
1829 {
return __is_const_known(__x._M_data0) && __is_const_known(__x._M_data1); }
1831 [[__gnu__::__always_inline__]]
1833 _M_concat_data([[maybe_unused]]
bool __do_sanitize =
false)
const
1835 return __vec_concat(_M_data0._M_concat_data(
false),
1836 __vec_zero_pad_to<
sizeof(_M_data0)>(
1837 _M_data1._M_concat_data(__do_sanitize)));
1840 template <
int _Size = _S_size,
int _Offset = 0,
typename _A0,
typename _Fp>
1841 [[__gnu__::__always_inline__]]
1842 static constexpr basic_vec
1843 _S_static_permute(
const basic_vec<value_type, _A0>& __x, _Fp&& __idxmap)
1846 _DataType0::template _S_static_permute<_Size, _Offset>(__x, __idxmap),
1847 _DataType1::template _S_static_permute<_Size, _Offset + _N0>(__x, __idxmap));
1850 template <
typename _Vp>
1851 [[__gnu__::__always_inline__]]
1853 _M_chunk() const noexcept
1855 constexpr int __n = _S_size / _Vp::_S_size;
1856 constexpr int __rem = _S_size % _Vp::_S_size;
1857 constexpr auto [...__is] = _IotaArray<__n>;
1858 if constexpr (__rem == 0)
1859 return array<_Vp, __n>{__extract_simd_at<_Vp>(cw<_Vp::_S_size * __is>,
1860 _M_data0, _M_data1)...};
1863 using _Rest = resize_t<__rem, _Vp>;
1864 return tuple(__extract_simd_at<_Vp>(cw<_Vp::_S_size * __is>, _M_data0, _M_data1)...,
1865 __extract_simd_at<_Rest>(cw<_Vp::_S_size * __n>, _M_data0, _M_data1));
1869 [[__gnu__::__always_inline__]]
1870 static constexpr const basic_vec&
1871 _S_concat(
const basic_vec& __x0)
noexcept
1874 template <
typename... _As>
1875 requires (
sizeof...(_As) >= 2)
1876 [[__gnu__::__always_inline__]]
1877 static constexpr basic_vec
1878 _S_concat(
const basic_vec<value_type, _As>&... __xs)
noexcept
1880 static_assert(_S_size == (_As::_S_size + ...));
1881 return _S_init(__extract_simd_at<_DataType0>(cw<0>, __xs...),
1882 __extract_simd_at<_DataType1>(cw<_N0>, __xs...));
1885 [[__gnu__::__always_inline__]]
1887 _M_reduce_to_half(
auto __binary_op)
const requires (_N0 == _N1)
1888 {
return __binary_op(_M_data0, _M_data1); }
1890 [[__gnu__::__always_inline__]]
1891 constexpr value_type
1892 _M_reduce_tail(
const auto& __rest,
auto __binary_op)
const
1894 if constexpr (__rest.size() > _S_size)
1896 auto [__a, __b] = __rest.template _M_chunk<basic_vec>();
1897 return __binary_op(*
this, __a)._M_reduce_tail(__b, __binary_op);
1899 else if constexpr (__rest.size() == _S_size)
1900 return __binary_op(*
this, __rest)._M_reduce(__binary_op);
1902 return _M_reduce_to_half(__binary_op)._M_reduce_tail(__rest, __binary_op);
1905 template <
typename _BinaryOp, _TargetTraits _Traits = {}>
1906 [[__gnu__::__always_inline__]]
1907 constexpr value_type
1908 _M_reduce(_BinaryOp __binary_op)
const
1910 if constexpr (_Traits.template _M_eval_as_f32<value_type>()
1911 && (is_same_v<_BinaryOp, plus<>>
1912 || is_same_v<_BinaryOp, multiplies<>>))
1913 return value_type(rebind_t<float, basic_vec>(*this)._M_reduce(__binary_op));
1915 else if constexpr (is_integral_v<value_type> &&
sizeof(value_type) == 1
1916 && is_same_v<
decltype(__binary_op), multiplies<>>)
1929 if constexpr (_DataType1::_S_is_scalar)
1930 return __binary_op(_DataType1(_M_data0._M_reduce(__binary_op)), _M_data1)[0];
1932 else if constexpr (_S_size % 2 == 0)
1934 using _V16 = resize_t<_S_size / 2, rebind_t<unsigned short, basic_vec>>;
1935 auto __a = __builtin_bit_cast(_V16, *
this);
1936 return __binary_op(__a, __a >> __CHAR_BIT__)._M_reduce(__binary_op);
1940 using _V16 = rebind_t<unsigned short, basic_vec>;
1941 return _V16(*this)._M_reduce(__binary_op);
1946 return _M_data0._M_reduce_tail(_M_data1, __binary_op);
1949 [[__gnu__::__always_inline__]]
1951 _M_isnan() const requires is_floating_point_v<value_type>
1952 {
return mask_type::_S_init(_M_data0._M_isnan(), _M_data1._M_isnan()); }
1954 [[__gnu__::__always_inline__]]
1956 _M_isinf() const requires is_floating_point_v<value_type>
1957 {
return mask_type::_S_init(_M_data0._M_isinf(), _M_data1._M_isinf()); }
1959 [[__gnu__::__always_inline__]]
1961 _M_isunordered(basic_vec __y)
const requires is_floating_point_v<value_type>
1963 return mask_type::_S_init(_M_data0._M_isunordered(__y._M_data0),
1964 _M_data1._M_isunordered(__y._M_data1));
1967 [[__gnu__::__always_inline__]]
1969 _M_abs() const requires signed_integral<value_type>
1970 {
return _S_init(_M_data0._M_abs(), _M_data1._M_abs()); }
1972 [[__gnu__::__always_inline__]]
1974 _M_fabs() const requires floating_point<value_type>
1975 {
return _S_init(_M_data0._M_fabs(), _M_data1._M_fabs()); }
1977 template <
typename _Up>
1978 [[__gnu__::__always_inline__]]
1979 static inline basic_vec
1980 _S_partial_load(
const _Up* __mem,
size_t __n)
1983 return _S_init(_DataType0(_LoadCtorTag(), __mem),
1984 _DataType1::_S_partial_load(__mem + _N0, __n - _N0));
1986 return _S_init(_DataType0::_S_partial_load(__mem, __n),
1990 template <
typename _Up, _ArchTraits _Traits = {}>
1991 static inline basic_vec
1992 _S_masked_load(
const _Up* __mem, mask_type __k)
1994 return _S_init(_DataType0::_S_masked_load(__mem, __k._M_data0),
1995 _DataType1::_S_masked_load(__mem + _N0, __k._M_data1));
1998 template <
typename _Up>
1999 [[__gnu__::__always_inline__]]
2001 _M_store(_Up* __mem)
const
2003 _M_data0._M_store(__mem);
2004 _M_data1._M_store(__mem + _N0);
2007 template <
typename _Up>
2008 [[__gnu__::__always_inline__]]
2010 _S_partial_store(
const basic_vec& __v, _Up* __mem,
size_t __n)
2014 __v._M_data0._M_store(__mem);
2015 _DataType1::_S_partial_store(__v._M_data1, __mem + _N0, __n - _N0);
2019 _DataType0::_S_partial_store(__v._M_data0, __mem, __n);
2023 template <
typename _Up>
2024 [[__gnu__::__always_inline__]]
2026 _S_masked_store(
const basic_vec& __v, _Up* __mem,
const mask_type& __k)
2028 _DataType0::_S_masked_store(__v._M_data0, __mem, __k._M_data0);
2029 _DataType1::_S_masked_store(__v._M_data1, __mem + _N0, __k._M_data1);
2032 basic_vec() =
default;
2035 using _NativeVecType = __vec_builtin_type<value_type, __bit_ceil(
unsigned(_S_size))>;
2037 [[__gnu__::__always_inline__]]
2039 basic_vec(
const _NativeVecType& __x)
2040 : _M_data0(_VecOps<__vec_builtin_type<value_type, _N0>>::_S_extract(__x)),
2041 _M_data1(_VecOps<__vec_builtin_type<value_type, __bit_ceil(unsigned(_N1))>>
2042 ::_S_extract(__x, integral_constant<int, _N0>()))
2045 [[__gnu__::__always_inline__]]
2047 operator _NativeVecType()
const
2048 {
return _M_concat_data(); }
2051 template <__broadcast_constructible<value_type> _Up>
2052 [[__gnu__::__always_inline__]]
2054 basic_vec(_Up&& __x) noexcept
2055 : _M_data0(
static_cast<value_type
>(__x)), _M_data1(
static_cast<value_type
>(__x))
2059 template <
typename _Up,
typename _UAbi>
2060 requires (_S_size == _UAbi::_S_size)
2061 && __explicitly_convertible_to<_Up, value_type>
2062 [[__gnu__::__always_inline__]]
2064 explicit(!__value_preserving_convertible_to<_Up, value_type>
2065 || __higher_rank_than<_Up, value_type>)
2066 basic_vec(
const basic_vec<_Up, _UAbi>& __x) noexcept
2067 : _M_data0(get<0>(chunk<_N0>(__x))),
2068 _M_data1(get<1>(chunk<_N0>(__x)))
2071 using _VecBase<_Tp, _Ap>::_VecBase;
2074 template <__simd_generator_invokable<value_type, _S_size> _Fp>
2075 [[__gnu__::__always_inline__]]
2077 basic_vec(_Fp&& __gen)
2078 : _M_data0(__gen), _M_data1([&] [[__gnu__::__always_inline__]] (auto __i) {
2079 return __gen(__simd_size_c<__i + _N0>);
2084 template <
typename _Up>
2085 [[__gnu__::__always_inline__]]
2087 basic_vec(_LoadCtorTag,
const _Up* __ptr)
2088 : _M_data0(_LoadCtorTag(), __ptr),
2089 _M_data1(_LoadCtorTag(), __ptr + _N0)
2092 template <ranges::contiguous_range _Rg,
typename... _Flags>
2093 requires ranges::__static_sized_range<_Rg>
2094 && __vectorizable<ranges::range_value_t<_Rg>>
2095 && __explicitly_convertible_to<ranges::range_value_t<_Rg>, value_type>
2097 basic_vec(_Rg&& __range, flags<_Flags...> __flags = {})
2098 requires (ranges::size(__range) == _S_size)
2099 : basic_vec(_LoadCtorTag(),
2100 __flags.template _S_adjust_pointer<basic_vec>(ranges::
data(__range)))
2102 static_assert(__loadstore_convertible_to<ranges::range_value_t<_Rg>, value_type,
2107 [[__gnu__::__always_inline__]]
2108 constexpr value_type
2109 operator[](__simd_size_type __i)
const
2111 __glibcxx_simd_precondition(__i >= 0 && __i < _S_size,
"subscript is out of bounds");
2112 if (__is_const_known(__i))
2113 return __i < _N0 ? _M_data0[__i] : _M_data1[__i - _N0];
2116 using _AliasingT [[__gnu__::__may_alias__]] = value_type;
2117 return reinterpret_cast<const _AliasingT*
>(
this)[__i];
2122 [[__gnu__::__always_inline__]]
2123 constexpr basic_vec&
2124 operator++() noexcept requires requires(value_type __a) { ++__a; }
2131 [[__gnu__::__always_inline__]]
2133 operator++(
int)
noexcept requires requires(value_type __a) { __a++; }
2135 basic_vec __r = *
this;
2141 [[__gnu__::__always_inline__]]
2142 constexpr basic_vec&
2143 operator--() noexcept requires requires(value_type __a) { --__a; }
2150 [[__gnu__::__always_inline__]]
2152 operator--(
int)
noexcept requires requires(value_type __a) { __a--; }
2154 basic_vec __r = *
this;
2160 [[__gnu__::__always_inline__]]
2162 operator!() const noexcept requires requires(value_type __a) { !__a; }
2163 {
return mask_type::_S_init(!_M_data0, !_M_data1); }
2165 [[__gnu__::__always_inline__]]
2167 operator+() const noexcept requires requires(value_type __a) { +__a; }
2170 [[__gnu__::__always_inline__]]
2172 operator-() const noexcept requires requires(value_type __a) { -__a; }
2173 {
return _S_init(-_M_data0, -_M_data1); }
2175 [[__gnu__::__always_inline__]]
2177 operator~() const noexcept requires requires(value_type __a) { ~__a; }
2178 {
return _S_init(~_M_data0, ~_M_data1); }
2181#define _GLIBCXX_SIMD_DEFINE_OP(sym) \
2182 [[__gnu__::__always_inline__]] \
2183 friend constexpr basic_vec& \
2184 operator sym##=(basic_vec& __x, const basic_vec& __y) _GLIBCXX_SIMD_NOEXCEPT \
2186 __x._M_data0 sym##= __y._M_data0; \
2187 __x._M_data1 sym##= __y._M_data1; \
2191 _GLIBCXX_SIMD_DEFINE_OP(+)
2192 _GLIBCXX_SIMD_DEFINE_OP(-)
2193 _GLIBCXX_SIMD_DEFINE_OP(*)
2194 _GLIBCXX_SIMD_DEFINE_OP(/)
2195 _GLIBCXX_SIMD_DEFINE_OP(%)
2196 _GLIBCXX_SIMD_DEFINE_OP(&)
2197 _GLIBCXX_SIMD_DEFINE_OP(|)
2198 _GLIBCXX_SIMD_DEFINE_OP(^)
2199 _GLIBCXX_SIMD_DEFINE_OP(<<)
2200 _GLIBCXX_SIMD_DEFINE_OP(>>)
2202#undef _GLIBCXX_SIMD_DEFINE_OP
2204 [[__gnu__::__always_inline__]]
2205 friend constexpr basic_vec&
2206 operator<<=(basic_vec& __x, __simd_size_type __y) _GLIBCXX_SIMD_NOEXCEPT
2207 requires requires(value_type __a, __simd_size_type __b) { __a << __b; }
2209 __x._M_data0 <<= __y;
2210 __x._M_data1 <<= __y;
2214 [[__gnu__::__always_inline__]]
2215 friend constexpr basic_vec&
2216 operator>>=(basic_vec& __x, __simd_size_type __y) _GLIBCXX_SIMD_NOEXCEPT
2217 requires requires(value_type __a, __simd_size_type __b) { __a >> __b; }
2219 __x._M_data0 >>= __y;
2220 __x._M_data1 >>= __y;
2225 [[__gnu__::__always_inline__]]
2226 friend constexpr mask_type
2227 operator==(
const basic_vec& __x,
const basic_vec& __y)
noexcept
2228 {
return mask_type::_S_init(__x._M_data0 == __y._M_data0, __x._M_data1 == __y._M_data1); }
2230 [[__gnu__::__always_inline__]]
2231 friend constexpr mask_type
2232 operator!=(
const basic_vec& __x,
const basic_vec& __y)
noexcept
2233 {
return mask_type::_S_init(__x._M_data0 != __y._M_data0, __x._M_data1 != __y._M_data1); }
2235 [[__gnu__::__always_inline__]]
2236 friend constexpr mask_type
2237 operator<(
const basic_vec& __x,
const basic_vec& __y)
noexcept
2238 {
return mask_type::_S_init(__x._M_data0 < __y._M_data0, __x._M_data1 < __y._M_data1); }
2240 [[__gnu__::__always_inline__]]
2241 friend constexpr mask_type
2242 operator<=(
const basic_vec& __x,
const basic_vec& __y)
noexcept
2243 {
return mask_type::_S_init(__x._M_data0 <= __y._M_data0, __x._M_data1 <= __y._M_data1); }
2245 [[__gnu__::__always_inline__]]
2246 friend constexpr mask_type
2247 operator>(
const basic_vec& __x,
const basic_vec& __y)
noexcept
2248 {
return mask_type::_S_init(__x._M_data0 > __y._M_data0, __x._M_data1 > __y._M_data1); }
2250 [[__gnu__::__always_inline__]]
2251 friend constexpr mask_type
2252 operator>=(
const basic_vec& __x,
const basic_vec& __y)
noexcept
2253 {
return mask_type::_S_init(__x._M_data0 >= __y._M_data0, __x._M_data1 >= __y._M_data1); }
2256 [[__gnu__::__always_inline__]]
2257 friend constexpr basic_vec
2258 __select_impl(
const mask_type& __k,
const basic_vec& __t,
const basic_vec& __f)
noexcept
2260 return _S_init(__select_impl(__k._M_data0, __t._M_data0, __f._M_data0),
2261 __select_impl(__k._M_data1, __t._M_data1, __f._M_data1));
2266 template <ranges::contiguous_range _Rg,
typename... _Ts>
2267 requires ranges::__static_sized_range<_Rg>
2268 basic_vec(_Rg&& __r, _Ts...)
2269 -> basic_vec<ranges::range_value_t<_Rg>,
2270 __deduce_abi_t<ranges::range_value_t<_Rg>,
2272 static_cast<__simd_size_type
>(ranges::size(__r))>>;
2274 static_cast<__simd_size_type
>(
decltype(std::span(__r))::extent)>>;
2277 template <
size_t _Bytes,
typename _Ap>
2278 basic_vec(basic_mask<_Bytes, _Ap>)
2279 -> basic_vec<__integer_from<_Bytes>,
2280 decltype(__abi_rebind<__integer_from<_Bytes>, basic_mask<_Bytes, _Ap>::size.value,
2284 template <__vectorizable _Tp>
2285 requires is_arithmetic_v<_Tp>
2286 inline constexpr _Tp
2287 __iota<_Tp> = _Tp();
2289 template <
typename _Tp,
typename _Ap>
2290 inline constexpr basic_vec<_Tp, _Ap>
2291 __iota<basic_vec<_Tp, _Ap>> = basic_vec<_Tp, _Ap>([](_Tp __i) -> _Tp {
2293 "iota object would overflow");
2297_GLIBCXX_END_NAMESPACE_VERSION
2300#pragma GCC diagnostic pop
constexpr bool operator<=(const duration< _Rep1, _Period1 > &__lhs, const duration< _Rep2, _Period2 > &__rhs)
constexpr bool operator>=(const duration< _Rep1, _Period1 > &__lhs, const duration< _Rep2, _Period2 > &__rhs)
constexpr bool operator<(const duration< _Rep1, _Period1 > &__lhs, const duration< _Rep2, _Period2 > &__rhs)
constexpr bool operator>(const duration< _Rep1, _Period1 > &__lhs, const duration< _Rep2, _Period2 > &__rhs)
constexpr complex< _Tp > operator-(const complex< _Tp > &__x, const complex< _Tp > &__y)
Return new complex value x minus y.
constexpr complex< _Tp > operator+(const complex< _Tp > &__x, const complex< _Tp > &__y)
Return new complex value x plus y.
bool is_sufficiently_aligned(_Tp *__ptr)
Is __ptr aligned to an _Align byte boundary?
ISO C++ entities toplevel namespace is std.
_Tp fabs(const std::complex< _Tp > &__z)
fabs(__z) TR1 8.1.8 [tr.c99.cmplx.fabs]
constexpr auto data(_Container &__cont) noexcept(noexcept(__cont.data())) -> decltype(__cont.data())
Return the data pointer of a container.
static constexpr _Tp max() noexcept
static constexpr _Tp infinity() noexcept