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_SIMD_H 115f757f3fSDimitry Andric #define _LIBCPP_EXPERIMENTAL___SIMD_SIMD_H 125f757f3fSDimitry Andric 135f757f3fSDimitry Andric #include <__type_traits/is_same.h> 145f757f3fSDimitry Andric #include <__type_traits/remove_cvref.h> 155f757f3fSDimitry Andric #include <__utility/forward.h> 165f757f3fSDimitry Andric #include <cstddef> 175f757f3fSDimitry Andric #include <experimental/__config> 185f757f3fSDimitry Andric #include <experimental/__simd/declaration.h> 195f757f3fSDimitry Andric #include <experimental/__simd/reference.h> 205f757f3fSDimitry Andric #include <experimental/__simd/traits.h> 215f757f3fSDimitry Andric #include <experimental/__simd/utility.h> 225f757f3fSDimitry Andric 235f757f3fSDimitry Andric #if _LIBCPP_STD_VER >= 17 && defined(_LIBCPP_ENABLE_EXPERIMENTAL) 245f757f3fSDimitry Andric 255f757f3fSDimitry Andric _LIBCPP_BEGIN_NAMESPACE_EXPERIMENTAL 265f757f3fSDimitry Andric inline namespace parallelism_v2 { 275f757f3fSDimitry Andric 285f757f3fSDimitry Andric // class template simd [simd.class] 295f757f3fSDimitry Andric // TODO: implement simd class 305f757f3fSDimitry Andric template <class _Tp, class _Abi> 315f757f3fSDimitry Andric class simd { 325f757f3fSDimitry Andric using _Impl = __simd_operations<_Tp, _Abi>; 335f757f3fSDimitry Andric using _Storage = typename _Impl::_SimdStorage; 345f757f3fSDimitry Andric 355f757f3fSDimitry Andric _Storage __s_; 365f757f3fSDimitry Andric 375f757f3fSDimitry Andric public: 385f757f3fSDimitry Andric using value_type = _Tp; 395f757f3fSDimitry Andric using reference = __simd_reference<_Tp, _Storage, value_type>; 405f757f3fSDimitry Andric using mask_type = simd_mask<_Tp, _Abi>; 415f757f3fSDimitry Andric using abi_type = _Abi; 425f757f3fSDimitry Andric 435f757f3fSDimitry Andric static _LIBCPP_HIDE_FROM_ABI constexpr size_t size() noexcept { return simd_size_v<value_type, abi_type>; } 445f757f3fSDimitry Andric 455f757f3fSDimitry Andric _LIBCPP_HIDE_FROM_ABI simd() noexcept = default; 465f757f3fSDimitry Andric 475f757f3fSDimitry Andric // broadcast constructor 485f757f3fSDimitry Andric template <class _Up, enable_if_t<__can_broadcast_v<value_type, __remove_cvref_t<_Up>>, int> = 0> 495f757f3fSDimitry Andric _LIBCPP_HIDE_FROM_ABI simd(_Up&& __v) noexcept : __s_(_Impl::__broadcast(static_cast<value_type>(__v))) {} 505f757f3fSDimitry Andric 515f757f3fSDimitry Andric // implicit type conversion constructor 525f757f3fSDimitry Andric template <class _Up, 535f757f3fSDimitry Andric enable_if_t<!is_same_v<_Up, _Tp> && is_same_v<abi_type, simd_abi::fixed_size<size()>> && 545f757f3fSDimitry Andric __is_non_narrowing_convertible_v<_Up, value_type>, 555f757f3fSDimitry Andric int> = 0> 565f757f3fSDimitry Andric _LIBCPP_HIDE_FROM_ABI simd(const simd<_Up, simd_abi::fixed_size<size()>>& __v) noexcept { 575f757f3fSDimitry Andric for (size_t __i = 0; __i < size(); __i++) { 585f757f3fSDimitry Andric (*this)[__i] = static_cast<value_type>(__v[__i]); 595f757f3fSDimitry Andric } 605f757f3fSDimitry Andric } 615f757f3fSDimitry Andric 625f757f3fSDimitry Andric // generator constructor 635f757f3fSDimitry Andric template <class _Generator, enable_if_t<__can_generate_v<value_type, _Generator, size()>, int> = 0> 645f757f3fSDimitry Andric explicit _LIBCPP_HIDE_FROM_ABI simd(_Generator&& __g) noexcept 655f757f3fSDimitry Andric : __s_(_Impl::__generate(std::forward<_Generator>(__g))) {} 665f757f3fSDimitry Andric 67*7a6dacacSDimitry Andric // load constructor 68*7a6dacacSDimitry Andric template <class _Up, class _Flags, enable_if_t<__is_vectorizable_v<_Up> && is_simd_flag_type_v<_Flags>, int> = 0> 69*7a6dacacSDimitry Andric _LIBCPP_HIDE_FROM_ABI simd(const _Up* __mem, _Flags) { 70*7a6dacacSDimitry Andric _Impl::__load(__s_, _Flags::template __apply<simd>(__mem)); 71*7a6dacacSDimitry Andric } 72*7a6dacacSDimitry Andric 735f757f3fSDimitry Andric // scalar access [simd.subscr] 745f757f3fSDimitry Andric _LIBCPP_HIDE_FROM_ABI reference operator[](size_t __i) noexcept { return reference(__s_, __i); } 755f757f3fSDimitry Andric _LIBCPP_HIDE_FROM_ABI value_type operator[](size_t __i) const noexcept { return __s_.__get(__i); } 765f757f3fSDimitry Andric }; 775f757f3fSDimitry Andric 785f757f3fSDimitry Andric template <class _Tp, class _Abi> 795f757f3fSDimitry Andric inline constexpr bool is_simd_v<simd<_Tp, _Abi>> = true; 805f757f3fSDimitry Andric 815f757f3fSDimitry Andric template <class _Tp> 825f757f3fSDimitry Andric using native_simd = simd<_Tp, simd_abi::native<_Tp>>; 835f757f3fSDimitry Andric 845f757f3fSDimitry Andric template <class _Tp, int _Np> 855f757f3fSDimitry Andric using fixed_size_simd = simd<_Tp, simd_abi::fixed_size<_Np>>; 865f757f3fSDimitry Andric 875f757f3fSDimitry Andric } // namespace parallelism_v2 885f757f3fSDimitry Andric _LIBCPP_END_NAMESPACE_EXPERIMENTAL 895f757f3fSDimitry Andric 905f757f3fSDimitry Andric #endif // _LIBCPP_STD_VER >= 17 && defined(_LIBCPP_ENABLE_EXPERIMENTAL) 915f757f3fSDimitry Andric #endif // _LIBCPP_EXPERIMENTAL___SIMD_SIMD_H 92