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