111cd02dfb91661c65134cac258cf5924270e9d2Dan Albert//===----------------------------------------------------------------------===//
211cd02dfb91661c65134cac258cf5924270e9d2Dan Albert//
311cd02dfb91661c65134cac258cf5924270e9d2Dan Albert//                     The LLVM Compiler Infrastructure
411cd02dfb91661c65134cac258cf5924270e9d2Dan Albert//
511cd02dfb91661c65134cac258cf5924270e9d2Dan Albert// This file is dual licensed under the MIT and the University of Illinois Open
611cd02dfb91661c65134cac258cf5924270e9d2Dan Albert// Source Licenses. See LICENSE.TXT for details.
711cd02dfb91661c65134cac258cf5924270e9d2Dan Albert//
811cd02dfb91661c65134cac258cf5924270e9d2Dan Albert//===----------------------------------------------------------------------===//
911cd02dfb91661c65134cac258cf5924270e9d2Dan Albert//
1011cd02dfb91661c65134cac258cf5924270e9d2Dan Albert// UNSUPPORTED: libcpp-has-no-threads
1111cd02dfb91661c65134cac258cf5924270e9d2Dan Albert
1211cd02dfb91661c65134cac258cf5924270e9d2Dan Albert// <atomic>
1311cd02dfb91661c65134cac258cf5924270e9d2Dan Albert
1411cd02dfb91661c65134cac258cf5924270e9d2Dan Albert// template <class Integral>
1511cd02dfb91661c65134cac258cf5924270e9d2Dan Albert//     Integral
1611cd02dfb91661c65134cac258cf5924270e9d2Dan Albert//     atomic_fetch_or_explicit(volatile atomic<Integral>* obj, Integral op);
1711cd02dfb91661c65134cac258cf5924270e9d2Dan Albert//
1811cd02dfb91661c65134cac258cf5924270e9d2Dan Albert// template <class Integral>
1911cd02dfb91661c65134cac258cf5924270e9d2Dan Albert//     Integral
2011cd02dfb91661c65134cac258cf5924270e9d2Dan Albert//     atomic_fetch_or_explicit(atomic<Integral>* obj, Integral op);
2111cd02dfb91661c65134cac258cf5924270e9d2Dan Albert
2211cd02dfb91661c65134cac258cf5924270e9d2Dan Albert#include <atomic>
2311cd02dfb91661c65134cac258cf5924270e9d2Dan Albert#include <type_traits>
2411cd02dfb91661c65134cac258cf5924270e9d2Dan Albert#include <cassert>
2511cd02dfb91661c65134cac258cf5924270e9d2Dan Albert
2611cd02dfb91661c65134cac258cf5924270e9d2Dan Albert#include "atomic_helpers.h"
2711cd02dfb91661c65134cac258cf5924270e9d2Dan Albert
2811cd02dfb91661c65134cac258cf5924270e9d2Dan Alberttemplate <class T>
2911cd02dfb91661c65134cac258cf5924270e9d2Dan Albertstruct TestFn {
3011cd02dfb91661c65134cac258cf5924270e9d2Dan Albert  void operator()() const {
3111cd02dfb91661c65134cac258cf5924270e9d2Dan Albert    {
3211cd02dfb91661c65134cac258cf5924270e9d2Dan Albert        typedef std::atomic<T> A;
3311cd02dfb91661c65134cac258cf5924270e9d2Dan Albert        A t;
3411cd02dfb91661c65134cac258cf5924270e9d2Dan Albert        std::atomic_init(&t, T(1));
3511cd02dfb91661c65134cac258cf5924270e9d2Dan Albert        assert(std::atomic_fetch_or_explicit(&t, T(2),
3611cd02dfb91661c65134cac258cf5924270e9d2Dan Albert               std::memory_order_seq_cst) == T(1));
3711cd02dfb91661c65134cac258cf5924270e9d2Dan Albert        assert(t == T(3));
3811cd02dfb91661c65134cac258cf5924270e9d2Dan Albert    }
3911cd02dfb91661c65134cac258cf5924270e9d2Dan Albert    {
4011cd02dfb91661c65134cac258cf5924270e9d2Dan Albert        typedef std::atomic<T> A;
4111cd02dfb91661c65134cac258cf5924270e9d2Dan Albert        volatile A t;
4211cd02dfb91661c65134cac258cf5924270e9d2Dan Albert        std::atomic_init(&t, T(3));
4311cd02dfb91661c65134cac258cf5924270e9d2Dan Albert        assert(std::atomic_fetch_or_explicit(&t, T(2),
4411cd02dfb91661c65134cac258cf5924270e9d2Dan Albert               std::memory_order_seq_cst) == T(3));
4511cd02dfb91661c65134cac258cf5924270e9d2Dan Albert        assert(t == T(3));
4611cd02dfb91661c65134cac258cf5924270e9d2Dan Albert    }
4711cd02dfb91661c65134cac258cf5924270e9d2Dan Albert  }
4811cd02dfb91661c65134cac258cf5924270e9d2Dan Albert};
4911cd02dfb91661c65134cac258cf5924270e9d2Dan Albert
5011cd02dfb91661c65134cac258cf5924270e9d2Dan Albertint main()
5111cd02dfb91661c65134cac258cf5924270e9d2Dan Albert{
5211cd02dfb91661c65134cac258cf5924270e9d2Dan Albert    TestEachIntegralType<TestFn>()();
5311cd02dfb91661c65134cac258cf5924270e9d2Dan Albert}
54