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 G> explicit simd(G&& gen);
15 
16 #include <experimental/simd>
17 #include <cstdint>
18 #include <cassert>
19 
20 #include "test_macros.h"
21 
22 namespace ex = std::experimental::parallelism_v2;
23 
24 template <class T, class... Args>
25 auto not_supported_simd128_ctor(Args&&... args) -> decltype(
26     ex::fixed_size_simd<T, 16 / sizeof(T)>(std::forward<Args>(args)...),
27     void()) = delete;
28 
29 template <class T>
30 void not_supported_simd128_ctor(...) {}
31 
32 template <class T, class... Args>
33 auto supported_simd128_ctor(Args&&... args) -> decltype(
34     ex::fixed_size_simd<T, 16 / sizeof(T)>(std::forward<Args>(args)...),
35     void()) {}
36 
37 template <class T>
38 void supported_simd128_ctor(...) = delete;
39 
40 struct identity {
41   template <size_t value>
42   int operator()(std::integral_constant<size_t, value>) const {
43     return value;
44   }
45 };
46 
47 void compile_generator() {
48   supported_simd128_ctor<int>(identity());
49   not_supported_simd128_ctor<int>([](int i) { return float(i); });
50   not_supported_simd128_ctor<int>([](intptr_t i) { return (int*)(i); });
51   not_supported_simd128_ctor<int>([](int* i) { return i; });
52 }
53 
54 struct limited_identity {
55   template <size_t value>
56   typename std::conditional<value <= 2, int32_t, int64_t>::type
57   operator()(std::integral_constant<size_t, value>) const {
58     return value;
59   }
60 };
61 
62 void compile_limited_identity() {
63   supported_simd128_ctor<int64_t>(limited_identity());
64   not_supported_simd128_ctor<int32_t>(limited_identity());
65 }
66 
67 template <typename SimdType>
68 void test_generator() {
69   {
70     SimdType a([](int i) { return i; });
71     assert(a[0] == 0);
72     assert(a[1] == 1);
73     assert(a[2] == 2);
74     assert(a[3] == 3);
75   }
76   {
77     SimdType a([](int i) { return 2 * i - 1; });
78     assert(a[0] == -1);
79     assert(a[1] == 1);
80     assert(a[2] == 3);
81     assert(a[3] == 5);
82   }
83 }
84 
85 int main(int, char**) {
86   // TODO: adjust the tests when this assertion fails.
87   assert(ex::native_simd<int32_t>::size() >= 4);
88   test_generator<ex::native_simd<int32_t>>();
89   test_generator<ex::fixed_size_simd<int32_t, 4>>();
90 
91   return 0;
92 }
93