xref: /llvm-project/libcxx/test/std/experimental/simd/simd.class/simd_ctor_load.pass.cpp (revision 5c663aa9ae4656c46d09914994859118759b426b)
1 //===----------------------------------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 // UNSUPPORTED: c++03, c++11, c++14
10 
11 // <experimental/simd>
12 //
13 // [simd.class]
14 // template<class U, class Flags> simd(const U* mem, Flags);
15 
16 #include "../test_utils.h"
17 
18 namespace ex = std::experimental::parallelism_v2;
19 
20 template <class T, class SimdAbi, std::size_t array_size>
21 struct ElementAlignedLoadCtorHelper {
22   template <class U>
operator ()ElementAlignedLoadCtorHelper23   void operator()() const {
24     U buffer[array_size];
25     for (size_t i = 0; i < array_size; ++i)
26       buffer[i] = static_cast<U>(i);
27     ex::simd<T, SimdAbi> origin_simd(buffer, ex::element_aligned_tag());
28     assert_simd_values_equal(origin_simd, buffer);
29   }
30 };
31 
32 template <class T, class SimdAbi, std::size_t array_size>
33 struct VectorAlignedLoadCtorHelper {
34   template <class U>
operator ()VectorAlignedLoadCtorHelper35   void operator()() const {
36     alignas(ex::memory_alignment_v<ex::simd<T, SimdAbi>, U>) U buffer[array_size];
37     for (size_t i = 0; i < array_size; ++i)
38       buffer[i] = static_cast<U>(i);
39     ex::simd<T, SimdAbi> origin_simd(buffer, ex::vector_aligned_tag());
40     assert_simd_values_equal(origin_simd, buffer);
41   }
42 };
43 
44 template <class T, class SimdAbi, std::size_t array_size>
45 struct OveralignedLoadCtorHelper {
46   template <class U>
operator ()OveralignedLoadCtorHelper47   void operator()() const {
48     alignas(bit_ceil(sizeof(U) + 1)) U buffer[array_size];
49     for (size_t i = 0; i < array_size; ++i)
50       buffer[i] = static_cast<U>(i);
51     ex::simd<T, SimdAbi> origin_simd(buffer, ex::overaligned_tag<bit_ceil(sizeof(U) + 1)>());
52     assert_simd_values_equal(origin_simd, buffer);
53   }
54 };
55 
56 template <class T, std::size_t>
57 struct CheckSimdLoadCtor {
58   template <class SimdAbi>
operator ()CheckSimdLoadCtor59   void operator()() {
60     constexpr std::size_t array_size = ex::simd_size_v<T, SimdAbi>;
61 
62     types::for_each(simd_test_types(), ElementAlignedLoadCtorHelper<T, SimdAbi, array_size>());
63     types::for_each(simd_test_types(), VectorAlignedLoadCtorHelper<T, SimdAbi, array_size>());
64     types::for_each(simd_test_types(), OveralignedLoadCtorHelper<T, SimdAbi, array_size>());
65   }
66 };
67 
68 template <class T, std::size_t>
69 struct CheckLoadCtorTraits {
70   template <class SimdAbi>
operator ()CheckLoadCtorTraits71   void operator()() {
72     // This function shall not participate in overload resolution unless
73     // is_simd_flag_type_v<Flags> is true, and
74     // U is a vectorizable type.
75     static_assert(std::is_constructible_v<ex::simd<T, SimdAbi>, const int*, ex::element_aligned_tag>);
76 
77     // is_simd_flag_type_v<Flags> is false
78     static_assert(!std::is_constructible_v<ex::simd<T, SimdAbi>, const int*, T>);
79     static_assert(!std::is_constructible_v<ex::simd<T, SimdAbi>, const int*, SimdAbi>);
80 
81     // U is not a vectorizable type.
82     static_assert(!std::is_constructible_v<ex::simd<T, SimdAbi>, const SimdAbi*, ex::element_aligned_tag>);
83     static_assert(
84         !std::is_constructible_v<ex::simd<T, SimdAbi>, const ex::element_aligned_tag*, ex::element_aligned_tag>);
85   }
86 };
87 
main(int,char **)88 int main(int, char**) {
89   test_all_simd_abi<CheckSimdLoadCtor>();
90   test_all_simd_abi<CheckLoadCtorTraits>();
91   return 0;
92 }
93