File: simd_ctor_broadcast.pass.cpp

package info (click to toggle)
llvm-toolchain-19 1%3A19.1.7-3~deb12u1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm-proposed-updates
  • size: 1,998,492 kB
  • sloc: cpp: 6,951,680; ansic: 1,486,157; asm: 913,598; python: 232,024; f90: 80,126; objc: 75,281; lisp: 37,276; pascal: 16,990; sh: 10,009; ml: 5,058; perl: 4,724; awk: 3,523; makefile: 3,167; javascript: 2,504; xml: 892; fortran: 664; cs: 573
file content (125 lines) | stat: -rw-r--r-- 4,119 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

// UNSUPPORTED: c++03, c++11, c++14
// XFAIL: target=powerpc{{.*}}le-unknown-linux-gnu

// <experimental/simd>
//
// [simd.class]
// template<class U> simd(U&& value) noexcept;

#include "../test_utils.h"

namespace ex = std::experimental::parallelism_v2;

template <class T, class SimdAbi, std::size_t array_size>
struct BroadCastHelper {
  const std::array<T, array_size>& expected_value;

  BroadCastHelper(const std::array<T, array_size>& value) : expected_value(value) {}

  template <class U>
  void operator()() const {
    if constexpr (is_non_narrowing_convertible_v<U, T>) {
      ex::simd<T, SimdAbi> simd_broadcast_from_vectorizable_type(static_cast<U>(3));
      assert_simd_values_equal<array_size>(simd_broadcast_from_vectorizable_type, expected_value);
    }
  }
};

template <class T, std::size_t>
struct CheckSimdBroadcastCtorFromVectorizedType {
  template <class SimdAbi>
  void operator()() {
    constexpr std::size_t array_size = ex::simd_size_v<T, SimdAbi>;
    std::array<T, array_size> expected_value;
    std::fill(expected_value.begin(), expected_value.end(), 3);

    types::for_each(simd_test_types(), BroadCastHelper<T, SimdAbi, array_size>(expected_value));
  }
};

template <class T>
class implicit_type {
  T val;

public:
  implicit_type(T v) : val(v) {}
  operator T() const { return val; }
};

template <class T, std::size_t>
struct CheckSimdBroadcastCtor {
  template <class SimdAbi>
  void operator()() {
    constexpr std::size_t array_size = ex::simd_size_v<T, SimdAbi>;
    std::array<T, array_size> expected_value;
    std::fill(expected_value.begin(), expected_value.end(), 3);

    implicit_type<T> implicit_convert_to_3(3);
    ex::simd<T, SimdAbi> simd_broadcast_from_implicit_type(std::move(implicit_convert_to_3));
    assert_simd_values_equal<array_size>(simd_broadcast_from_implicit_type, expected_value);

    ex::simd<T, SimdAbi> simd_broadcast_from_int(3);
    assert_simd_values_equal<array_size>(simd_broadcast_from_int, expected_value);

    if constexpr (std::is_unsigned_v<T>) {
      ex::simd<T, SimdAbi> simd_broadcast_from_uint(3u);
      assert_simd_values_equal<array_size>(simd_broadcast_from_uint, expected_value);
    }
  }
};

template <class T>
class no_implicit_type {
  T val;

public:
  no_implicit_type(T v) : val(v) {}
};

template <class U, class T, class SimdAbi = ex::simd_abi::compatible<T>, class = void>
struct has_broadcast_ctor : std::false_type {};

template <class U, class T, class SimdAbi>
struct has_broadcast_ctor<U, T, SimdAbi, std::void_t<decltype(ex::simd<T, SimdAbi>(std::declval<U>()))>>
    : std::true_type {};

template <class T, class SimdAbi>
struct CheckBroadcastCtorTraitsHelper {
  template <class U>
  void operator()() const {
    if constexpr (std::is_same_v<U, int>)
      static_assert(has_broadcast_ctor<U, T, SimdAbi>::value);
    else if constexpr (std::is_same_v<U, unsigned int> && std::is_unsigned_v<T>)
      static_assert(has_broadcast_ctor<U, T, SimdAbi>::value);
    else if constexpr (is_non_narrowing_convertible_v<U, T>)
      static_assert(has_broadcast_ctor<U, T, SimdAbi>::value);
    else
      static_assert(!has_broadcast_ctor<U, T, SimdAbi>::value);
  }
};

template <class T, std::size_t>
struct CheckBroadcastCtorTraits {
  template <class SimdAbi>
  void operator()() {
    types::for_each(simd_test_types(), CheckBroadcastCtorTraitsHelper<T, SimdAbi>());

    static_assert(!has_broadcast_ctor<no_implicit_type<T>, T, SimdAbi>::value);
    static_assert(has_broadcast_ctor<implicit_type<T>, T, SimdAbi>::value);
  }
};

int main(int, char**) {
  test_all_simd_abi<CheckSimdBroadcastCtorFromVectorizedType>();
  test_all_simd_abi<CheckSimdBroadcastCtor>();
  test_all_simd_abi<CheckBroadcastCtorTraits>();
  return 0;
}