| 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 | // XFAIL: !has-64-bit-atomics |
| 11 | |
| 12 | // <atomic> |
| 13 | // |
| 14 | // template <class T> |
| 15 | // class atomic; |
| 16 | // |
| 17 | // static constexpr bool is_always_lock_free; |
| 18 | |
| 19 | // Ignore diagnostic about vector types changing the ABI on some targets, since |
| 20 | // that is irrelevant for this test. |
| 21 | // ADDITIONAL_COMPILE_FLAGS(gcc-style-warnings): -Wno-psabi |
| 22 | |
| 23 | #include <atomic> |
| 24 | #include <cassert> |
| 25 | #include <concepts> |
| 26 | #include <cstddef> |
| 27 | |
| 28 | #include "test_macros.h" |
| 29 | #include "atomic_helpers.h" |
| 30 | |
| 31 | template <typename T> |
| 32 | void check_always_lock_free(std::atomic<T> const& a) { |
| 33 | using InfoT = LockFreeStatusInfo<T>; |
| 34 | |
| 35 | constexpr auto is_always_lock_free = std::atomic<T>::is_always_lock_free; |
| 36 | ASSERT_SAME_TYPE(decltype(is_always_lock_free), bool const); |
| 37 | |
| 38 | // If we know the status of T for sure, validate the exact result of the function. |
| 39 | if constexpr (InfoT::status_known) { |
| 40 | constexpr LockFreeStatus known_status = InfoT::value; |
| 41 | if constexpr (known_status == LockFreeStatus::always) { |
| 42 | static_assert(is_always_lock_free, "is_always_lock_free is inconsistent with known lock-free status" ); |
| 43 | assert(a.is_lock_free() && "is_lock_free() is inconsistent with known lock-free status" ); |
| 44 | } else if constexpr (known_status == LockFreeStatus::never) { |
| 45 | static_assert(!is_always_lock_free, "is_always_lock_free is inconsistent with known lock-free status" ); |
| 46 | assert(!a.is_lock_free() && "is_lock_free() is inconsistent with known lock-free status" ); |
| 47 | } else { |
| 48 | assert(a.is_lock_free() || !a.is_lock_free()); // This is kinda dumb, but we might as well call the function once. |
| 49 | } |
| 50 | } |
| 51 | |
| 52 | // In all cases, also sanity-check it based on the implication always-lock-free => lock-free. |
| 53 | if (is_always_lock_free) { |
| 54 | auto is_lock_free = a.is_lock_free(); |
| 55 | ASSERT_SAME_TYPE(decltype(is_lock_free), bool); |
| 56 | assert(is_lock_free); |
| 57 | } |
| 58 | ASSERT_NOEXCEPT(a.is_lock_free()); |
| 59 | } |
| 60 | |
| 61 | #define CHECK_ALWAYS_LOCK_FREE(T) \ |
| 62 | do { \ |
| 63 | typedef T type; \ |
| 64 | type obj{}; \ |
| 65 | std::atomic<type> a(obj); \ |
| 66 | check_always_lock_free(a); \ |
| 67 | } while (0) |
| 68 | |
| 69 | void test() { |
| 70 | char c = 'x'; |
| 71 | check_always_lock_free(a: std::atomic<char>(c)); |
| 72 | |
| 73 | int i = 0; |
| 74 | check_always_lock_free(a: std::atomic<int>(i)); |
| 75 | |
| 76 | float f = 0.f; |
| 77 | check_always_lock_free(a: std::atomic<float>(f)); |
| 78 | |
| 79 | int* p = &i; |
| 80 | check_always_lock_free(a: std::atomic<int*>(p)); |
| 81 | |
| 82 | CHECK_ALWAYS_LOCK_FREE(bool); |
| 83 | CHECK_ALWAYS_LOCK_FREE(char); |
| 84 | CHECK_ALWAYS_LOCK_FREE(signed char); |
| 85 | CHECK_ALWAYS_LOCK_FREE(unsigned char); |
| 86 | #if TEST_STD_VER > 17 && defined(__cpp_char8_t) |
| 87 | CHECK_ALWAYS_LOCK_FREE(char8_t); |
| 88 | #endif |
| 89 | CHECK_ALWAYS_LOCK_FREE(char16_t); |
| 90 | CHECK_ALWAYS_LOCK_FREE(char32_t); |
| 91 | CHECK_ALWAYS_LOCK_FREE(wchar_t); |
| 92 | CHECK_ALWAYS_LOCK_FREE(short); |
| 93 | CHECK_ALWAYS_LOCK_FREE(unsigned short); |
| 94 | CHECK_ALWAYS_LOCK_FREE(int); |
| 95 | CHECK_ALWAYS_LOCK_FREE(unsigned int); |
| 96 | CHECK_ALWAYS_LOCK_FREE(long); |
| 97 | CHECK_ALWAYS_LOCK_FREE(unsigned long); |
| 98 | CHECK_ALWAYS_LOCK_FREE(long long); |
| 99 | CHECK_ALWAYS_LOCK_FREE(unsigned long long); |
| 100 | CHECK_ALWAYS_LOCK_FREE(std::nullptr_t); |
| 101 | CHECK_ALWAYS_LOCK_FREE(void*); |
| 102 | CHECK_ALWAYS_LOCK_FREE(float); |
| 103 | CHECK_ALWAYS_LOCK_FREE(double); |
| 104 | CHECK_ALWAYS_LOCK_FREE(long double); |
| 105 | #if __has_attribute(vector_size) && defined(_LIBCPP_VERSION) |
| 106 | CHECK_ALWAYS_LOCK_FREE(int __attribute__((vector_size(1 * sizeof(int))))); |
| 107 | CHECK_ALWAYS_LOCK_FREE(int __attribute__((vector_size(2 * sizeof(int))))); |
| 108 | CHECK_ALWAYS_LOCK_FREE(int __attribute__((vector_size(4 * sizeof(int))))); |
| 109 | CHECK_ALWAYS_LOCK_FREE(int __attribute__((vector_size(16 * sizeof(int))))); |
| 110 | CHECK_ALWAYS_LOCK_FREE(int __attribute__((vector_size(32 * sizeof(int))))); |
| 111 | CHECK_ALWAYS_LOCK_FREE(float __attribute__((vector_size(1 * sizeof(float))))); |
| 112 | CHECK_ALWAYS_LOCK_FREE(float __attribute__((vector_size(2 * sizeof(float))))); |
| 113 | CHECK_ALWAYS_LOCK_FREE(float __attribute__((vector_size(4 * sizeof(float))))); |
| 114 | CHECK_ALWAYS_LOCK_FREE(float __attribute__((vector_size(16 * sizeof(float))))); |
| 115 | CHECK_ALWAYS_LOCK_FREE(float __attribute__((vector_size(32 * sizeof(float))))); |
| 116 | CHECK_ALWAYS_LOCK_FREE(double __attribute__((vector_size(1 * sizeof(double))))); |
| 117 | CHECK_ALWAYS_LOCK_FREE(double __attribute__((vector_size(2 * sizeof(double))))); |
| 118 | CHECK_ALWAYS_LOCK_FREE(double __attribute__((vector_size(4 * sizeof(double))))); |
| 119 | CHECK_ALWAYS_LOCK_FREE(double __attribute__((vector_size(16 * sizeof(double))))); |
| 120 | CHECK_ALWAYS_LOCK_FREE(double __attribute__((vector_size(32 * sizeof(double))))); |
| 121 | #endif // __has_attribute(vector_size) && defined(_LIBCPP_VERSION) |
| 122 | CHECK_ALWAYS_LOCK_FREE(struct Empty{}); |
| 123 | CHECK_ALWAYS_LOCK_FREE(struct OneInt { int i; }); |
| 124 | CHECK_ALWAYS_LOCK_FREE(struct IntArr2 { int i[2]; }); |
| 125 | CHECK_ALWAYS_LOCK_FREE(struct FloatArr3 { float i[3]; }); |
| 126 | CHECK_ALWAYS_LOCK_FREE(struct LLIArr2 { long long int i[2]; }); |
| 127 | CHECK_ALWAYS_LOCK_FREE(struct LLIArr4 { long long int i[4]; }); |
| 128 | CHECK_ALWAYS_LOCK_FREE(struct LLIArr8 { long long int i[8]; }); |
| 129 | CHECK_ALWAYS_LOCK_FREE(struct LLIArr16 { long long int i[16]; }); |
| 130 | CHECK_ALWAYS_LOCK_FREE(struct Padding { |
| 131 | char c; /* padding */ |
| 132 | long long int i; |
| 133 | }); |
| 134 | CHECK_ALWAYS_LOCK_FREE(union IntFloat { |
| 135 | int i; |
| 136 | float f; |
| 137 | }); |
| 138 | CHECK_ALWAYS_LOCK_FREE(enum class CharEnumClass : char{foo}); |
| 139 | |
| 140 | // C macro and static constexpr must be consistent. |
| 141 | enum class CharEnumClass : char { foo }; |
| 142 | static_assert(std::atomic<bool>::is_always_lock_free == (2 == ATOMIC_BOOL_LOCK_FREE), "" ); |
| 143 | static_assert(std::atomic<char>::is_always_lock_free == (2 == ATOMIC_CHAR_LOCK_FREE), "" ); |
| 144 | static_assert(std::atomic<CharEnumClass>::is_always_lock_free == (2 == ATOMIC_CHAR_LOCK_FREE), "" ); |
| 145 | static_assert(std::atomic<signed char>::is_always_lock_free == (2 == ATOMIC_CHAR_LOCK_FREE), "" ); |
| 146 | static_assert(std::atomic<unsigned char>::is_always_lock_free == (2 == ATOMIC_CHAR_LOCK_FREE), "" ); |
| 147 | #if TEST_STD_VER > 17 && defined(__cpp_char8_t) |
| 148 | static_assert(std::atomic<char8_t>::is_always_lock_free == (2 == ATOMIC_CHAR8_T_LOCK_FREE), "" ); |
| 149 | #endif |
| 150 | static_assert(std::atomic<char16_t>::is_always_lock_free == (2 == ATOMIC_CHAR16_T_LOCK_FREE), "" ); |
| 151 | static_assert(std::atomic<char32_t>::is_always_lock_free == (2 == ATOMIC_CHAR32_T_LOCK_FREE), "" ); |
| 152 | static_assert(std::atomic<wchar_t>::is_always_lock_free == (2 == ATOMIC_WCHAR_T_LOCK_FREE), "" ); |
| 153 | static_assert(std::atomic<short>::is_always_lock_free == (2 == ATOMIC_SHORT_LOCK_FREE), "" ); |
| 154 | static_assert(std::atomic<unsigned short>::is_always_lock_free == (2 == ATOMIC_SHORT_LOCK_FREE), "" ); |
| 155 | static_assert(std::atomic<int>::is_always_lock_free == (2 == ATOMIC_INT_LOCK_FREE), "" ); |
| 156 | static_assert(std::atomic<unsigned int>::is_always_lock_free == (2 == ATOMIC_INT_LOCK_FREE), "" ); |
| 157 | static_assert(std::atomic<long>::is_always_lock_free == (2 == ATOMIC_LONG_LOCK_FREE), "" ); |
| 158 | static_assert(std::atomic<unsigned long>::is_always_lock_free == (2 == ATOMIC_LONG_LOCK_FREE), "" ); |
| 159 | static_assert(std::atomic<long long>::is_always_lock_free == (2 == ATOMIC_LLONG_LOCK_FREE), "" ); |
| 160 | static_assert(std::atomic<unsigned long long>::is_always_lock_free == (2 == ATOMIC_LLONG_LOCK_FREE), "" ); |
| 161 | static_assert(std::atomic<void*>::is_always_lock_free == (2 == ATOMIC_POINTER_LOCK_FREE), "" ); |
| 162 | static_assert(std::atomic<std::nullptr_t>::is_always_lock_free == (2 == ATOMIC_POINTER_LOCK_FREE), "" ); |
| 163 | |
| 164 | #if TEST_STD_VER >= 20 |
| 165 | static_assert(std::atomic_signed_lock_free::is_always_lock_free, "" ); |
| 166 | static_assert(std::atomic_unsigned_lock_free::is_always_lock_free, "" ); |
| 167 | #endif |
| 168 | } |
| 169 | |
| 170 | int main(int, char**) { |
| 171 | test(); |
| 172 | return 0; |
| 173 | } |
| 174 | |