09e3a36058
This patch adds a large number of missing includes in the libc++ headers and the test suite. Those were found as part of the effort to move towards a mostly monolithic top-level std module.
115 lines
3.5 KiB
C++
115 lines
3.5 KiB
C++
//
|
|
// 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, c++17
|
|
// XFAIL: !has-64-bit-atomics
|
|
|
|
// integral-type fetch_sub(integral-type, memory_order = memory_order::seq_cst) const noexcept;
|
|
// floating-point-type fetch_sub(floating-point-type, memory_order = memory_order::seq_cst) const noexcept;
|
|
// T* fetch_sub(difference_type, memory_order = memory_order::seq_cst) const noexcept;
|
|
|
|
#include <atomic>
|
|
#include <cassert>
|
|
#include <concepts>
|
|
#include <type_traits>
|
|
#include <utility>
|
|
|
|
#include "atomic_helpers.h"
|
|
#include "test_helper.h"
|
|
#include "test_macros.h"
|
|
|
|
template <typename T>
|
|
concept has_fetch_sub = requires {
|
|
std::declval<T const>().fetch_sub(std::declval<T>());
|
|
std::declval<T const>().fetch_sub(std::declval<T>(), std::declval<std::memory_order>());
|
|
};
|
|
|
|
template <typename T>
|
|
struct TestDoesNotHaveFetchSub {
|
|
void operator()() const { static_assert(!has_fetch_sub<std::atomic_ref<T>>); }
|
|
};
|
|
|
|
template <typename T>
|
|
struct TestFetchSub {
|
|
void operator()() const {
|
|
if constexpr (std::is_arithmetic_v<T>) {
|
|
T x(T(7));
|
|
std::atomic_ref<T> const a(x);
|
|
|
|
{
|
|
std::same_as<T> decltype(auto) y = a.fetch_sub(T(4));
|
|
assert(y == T(7));
|
|
assert(x == T(3));
|
|
ASSERT_NOEXCEPT(a.fetch_sub(T(0)));
|
|
}
|
|
|
|
{
|
|
std::same_as<T> decltype(auto) y = a.fetch_sub(T(2), std::memory_order_relaxed);
|
|
assert(y == T(3));
|
|
assert(x == T(1));
|
|
ASSERT_NOEXCEPT(a.fetch_sub(T(0), std::memory_order_relaxed));
|
|
}
|
|
} else if constexpr (std::is_pointer_v<T>) {
|
|
using U = std::remove_pointer_t<T>;
|
|
U t[9] = {};
|
|
T p{&t[7]};
|
|
std::atomic_ref<T> const a(p);
|
|
|
|
{
|
|
std::same_as<T> decltype(auto) y = a.fetch_sub(4);
|
|
assert(y == &t[7]);
|
|
assert(a == &t[3]);
|
|
ASSERT_NOEXCEPT(a.fetch_sub(0));
|
|
}
|
|
|
|
{
|
|
std::same_as<T> decltype(auto) y = a.fetch_sub(2, std::memory_order_relaxed);
|
|
assert(y == &t[3]);
|
|
assert(a == &t[1]);
|
|
ASSERT_NOEXCEPT(a.fetch_sub(0, std::memory_order_relaxed));
|
|
}
|
|
} else {
|
|
static_assert(std::is_void_v<T>);
|
|
}
|
|
|
|
// memory_order::release
|
|
{
|
|
auto fetch_sub = [](std::atomic_ref<T> const& x, T old_val, T new_val) {
|
|
x.fetch_sub(old_val - new_val, std::memory_order::release);
|
|
};
|
|
auto load = [](std::atomic_ref<T> const& x) { return x.load(std::memory_order::acquire); };
|
|
test_acquire_release<T>(fetch_sub, load);
|
|
}
|
|
|
|
// memory_order::seq_cst
|
|
{
|
|
auto fetch_sub_no_arg = [](std::atomic_ref<T> const& x, T old_val, T new_val) { x.fetch_sub(old_val - new_val); };
|
|
auto fetch_sub_with_order = [](std::atomic_ref<T> const& x, T old_val, T new_val) {
|
|
x.fetch_sub(old_val - new_val, std::memory_order::seq_cst);
|
|
};
|
|
auto load = [](std::atomic_ref<T> const& x) { return x.load(); };
|
|
test_seq_cst<T>(fetch_sub_no_arg, load);
|
|
test_seq_cst<T>(fetch_sub_with_order, load);
|
|
}
|
|
}
|
|
};
|
|
|
|
int main(int, char**) {
|
|
TestEachIntegralType<TestFetchSub>()();
|
|
|
|
TestFetchSub<float>()();
|
|
TestFetchSub<double>()();
|
|
|
|
TestEachPointerType<TestFetchSub>()();
|
|
|
|
TestDoesNotHaveFetchSub<bool>()();
|
|
TestDoesNotHaveFetchSub<UserAtomicType>()();
|
|
TestDoesNotHaveFetchSub<LargeUserAtomicType>()();
|
|
|
|
return 0;
|
|
}
|