xref: /llvm-project/libcxx/test/std/atomics/atomics.ref/store.pass.cpp (revision 42ba740afffa16f991be6aa36626bd872d41ebc0)
1*42ba740aSDamien L-G //
2*42ba740aSDamien L-G // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
3*42ba740aSDamien L-G // See https://llvm.org/LICENSE.txt for license information.
4*42ba740aSDamien L-G // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
5*42ba740aSDamien L-G //
6*42ba740aSDamien L-G //===----------------------------------------------------------------------===//
7*42ba740aSDamien L-G 
8*42ba740aSDamien L-G // UNSUPPORTED: c++03, c++11, c++14, c++17
9*42ba740aSDamien L-G // XFAIL: !has-64-bit-atomics
10*42ba740aSDamien L-G // XFAIL: !has-1024-bit-atomics
11*42ba740aSDamien L-G 
12*42ba740aSDamien L-G // void store(T, memory_order = memory_order::seq_cst) const noexcept;
13*42ba740aSDamien L-G 
14*42ba740aSDamien L-G #include <atomic>
15*42ba740aSDamien L-G #include <cassert>
16*42ba740aSDamien L-G #include <type_traits>
17*42ba740aSDamien L-G 
18*42ba740aSDamien L-G #include "atomic_helpers.h"
19*42ba740aSDamien L-G #include "test_helper.h"
20*42ba740aSDamien L-G #include "test_macros.h"
21*42ba740aSDamien L-G 
22*42ba740aSDamien L-G template <typename T>
23*42ba740aSDamien L-G struct TestStore {
operator ()TestStore24*42ba740aSDamien L-G   void operator()() const {
25*42ba740aSDamien L-G     T x(T(1));
26*42ba740aSDamien L-G     std::atomic_ref<T> const a(x);
27*42ba740aSDamien L-G 
28*42ba740aSDamien L-G     a.store(T(2));
29*42ba740aSDamien L-G     assert(x == T(2));
30*42ba740aSDamien L-G     ASSERT_NOEXCEPT(a.store(T(1)));
31*42ba740aSDamien L-G 
32*42ba740aSDamien L-G     a.store(T(3), std::memory_order_seq_cst);
33*42ba740aSDamien L-G     assert(x == T(3));
34*42ba740aSDamien L-G     ASSERT_NOEXCEPT(a.store(T(0), std::memory_order_seq_cst));
35*42ba740aSDamien L-G 
36*42ba740aSDamien L-G     // TODO memory_order::relaxed
37*42ba740aSDamien L-G 
38*42ba740aSDamien L-G     // memory_order::seq_cst
39*42ba740aSDamien L-G     {
40*42ba740aSDamien L-G       auto store_no_arg     = [](std::atomic_ref<T> const& y, T, T new_val) { y.store(new_val); };
41*42ba740aSDamien L-G       auto store_with_order = [](std::atomic_ref<T> const& y, T, T new_val) {
42*42ba740aSDamien L-G         y.store(new_val, std::memory_order::seq_cst);
43*42ba740aSDamien L-G       };
44*42ba740aSDamien L-G       auto load = [](std::atomic_ref<T> const& y) { return y.load(); };
45*42ba740aSDamien L-G       test_seq_cst<T>(store_no_arg, load);
46*42ba740aSDamien L-G       test_seq_cst<T>(store_with_order, load);
47*42ba740aSDamien L-G     }
48*42ba740aSDamien L-G 
49*42ba740aSDamien L-G     // memory_order::release
50*42ba740aSDamien L-G     {
51*42ba740aSDamien L-G       auto store = [](std::atomic_ref<T> const& y, T, T new_val) { y.store(new_val, std::memory_order::release); };
52*42ba740aSDamien L-G       auto load  = [](std::atomic_ref<T> const& y) { return y.load(std::memory_order::acquire); };
53*42ba740aSDamien L-G       test_acquire_release<T>(store, load);
54*42ba740aSDamien L-G     }
55*42ba740aSDamien L-G   }
56*42ba740aSDamien L-G };
57*42ba740aSDamien L-G 
main(int,char **)58*42ba740aSDamien L-G int main(int, char**) {
59*42ba740aSDamien L-G   TestEachAtomicType<TestStore>()();
60*42ba740aSDamien L-G   return 0;
61*42ba740aSDamien L-G }
62