File: simd_mask_copy.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 (127 lines) | stat: -rw-r--r-- 5,262 bytes parent folder | download | duplicates (7)
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
126
127
//===----------------------------------------------------------------------===//
//
// 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

// <experimental/simd>
//
// [simd.class]
// template<class Flags> void copy_from(const value_type* mem, Flags);
// template<class Flags> void copy_to(value_type* mem, Flags);

#include "../test_utils.h"

namespace ex = std::experimental::parallelism_v2;

template <class T, std::size_t>
struct CheckSimdMaskCopyFrom {
  template <class SimdAbi>
  void operator()() {
    constexpr std::size_t array_size = ex::simd_size_v<T, SimdAbi>;

    // element aligned tag
    constexpr std::size_t element_alignas_size = alignof(bool);
    alignas(element_alignas_size) bool element_buffer[array_size];
    for (size_t i = 0; i < array_size; ++i)
      element_buffer[i] = static_cast<bool>(i % 2);
    ex::simd_mask<T, SimdAbi> element_mask;
    element_mask.copy_from(element_buffer, ex::element_aligned_tag());
    assert_simd_mask_values_equal(element_mask, element_buffer);

    // vector aligned tag
    constexpr std::size_t vector_alignas_size = ex::memory_alignment_v<ex::simd_mask<T, SimdAbi>>;
    alignas(vector_alignas_size) bool vector_buffer[array_size];
    for (size_t i = 0; i < array_size; ++i)
      vector_buffer[i] = static_cast<bool>(i % 2);
    ex::simd_mask<T, SimdAbi> vector_mask;
    vector_mask.copy_from(vector_buffer, ex::vector_aligned_tag());
    assert_simd_mask_values_equal(vector_mask, vector_buffer);

    // overaligned tag
    constexpr std::size_t over_alignas_size = bit_ceil(sizeof(bool) + 1);
    alignas(over_alignas_size) bool overaligned_buffer[array_size];
    for (size_t i = 0; i < array_size; ++i)
      overaligned_buffer[i] = static_cast<bool>(i % 2);
    ex::simd_mask<T, SimdAbi> overaligned_mask;
    overaligned_mask.copy_from(overaligned_buffer, ex::overaligned_tag<over_alignas_size>());
    assert_simd_mask_values_equal(overaligned_mask, overaligned_buffer);
  }
};

template <class T, std::size_t>
struct CheckSimdMaskCopyTo {
  template <class SimdAbi>
  void operator()() {
    constexpr std::size_t array_size = ex::simd_size_v<T, SimdAbi>;

    // element aligned tag
    constexpr std::size_t element_alignas_size = alignof(bool);
    alignas(element_alignas_size) bool element_buffer[array_size];
    ex::simd_mask<T, SimdAbi> element_mask(true);
    element_mask.copy_to(element_buffer, ex::element_aligned_tag());
    assert_simd_mask_values_equal(element_mask, element_buffer);

    // vector aligned tag
    constexpr std::size_t vector_alignas_size = ex::memory_alignment_v<ex::simd_mask<T, SimdAbi>>;
    alignas(vector_alignas_size) bool vector_buffer[array_size];
    ex::simd_mask<T, SimdAbi> vector_mask(false);
    vector_mask.copy_to(vector_buffer, ex::vector_aligned_tag());
    assert_simd_mask_values_equal(vector_mask, vector_buffer);

    // overaligned tag
    constexpr std::size_t over_alignas_size = bit_ceil(sizeof(bool) + 1);
    alignas(over_alignas_size) bool overaligned_buffer[array_size];
    ex::simd_mask<T, SimdAbi> overaligned_mask(true);
    overaligned_mask.copy_to(overaligned_buffer, ex::overaligned_tag<over_alignas_size>());
    assert_simd_mask_values_equal(overaligned_mask, overaligned_buffer);
  }
};

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

template <class T, class Flags, class SimdAbi>
struct has_copy_from<T,
                     Flags,
                     SimdAbi,
                     std::void_t<decltype(std::declval<ex::simd_mask<T, SimdAbi>>().copy_from(
                         std::declval<const bool*>(), std::declval<Flags>()))>> : std::true_type {};

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

template <class T, class Flags, class SimdAbi>
struct has_copy_to<T,
                   Flags,
                   SimdAbi,
                   std::void_t<decltype(std::declval<ex::simd_mask<T, SimdAbi>>().copy_to(
                       std::declval<bool*>(), std::declval<Flags>()))>> : std::true_type {};

template <class T, std::size_t>
struct CheckSimdMaskCopyTraits {
  template <class SimdAbi>
  void operator()() {
    // These functions shall not participate in overload resolution unless
    // is_simd_flag_type_v<Flags> is true
    static_assert(has_copy_from<T, ex::element_aligned_tag, SimdAbi>::value);
    static_assert(has_copy_to<T, ex::element_aligned_tag, SimdAbi>::value);

    // is_simd_flag_type_v<Flags> is false
    static_assert(!has_copy_from<T, T, SimdAbi>::value);
    static_assert(!has_copy_to<T, T, SimdAbi>::value);
    static_assert(!has_copy_from<T, SimdAbi, SimdAbi>::value);
    static_assert(!has_copy_to<T, SimdAbi, SimdAbi>::value);
  }
};

int main(int, char**) {
  test_all_simd_abi<CheckSimdMaskCopyFrom>();
  test_all_simd_abi<CheckSimdMaskCopyTo>();
  test_all_simd_abi<CheckSimdMaskCopyTraits>();
  return 0;
}