/usr/include/TH/THAtomic.h is in libtorch-th-dev 0~20170926-g89ede3b-2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 | #ifndef TH_ATOMIC_INC
#define TH_ATOMIC_INC
#include "THGeneral.h"
/******************************************************************************
* Atomic operations for TH
* Five backends are integrated:
* - C11 atomic operations
* - MSVC intrinsics
* - GCC intrinsics
* - Pthread if none of the above is available
* - Unsafe mode in none of the above is available
******************************************************************************/
/******************************************************************************
* all-purpose functions
******************************************************************************/
/*
* *a = newvalue
*/
TH_API void THAtomicSet(int volatile *a, int newvalue);
/*
* return *a
*/
TH_API int THAtomicGet(int volatile *a);
/*
* *a += value,
* return previous *a
*/
TH_API int THAtomicAdd(int volatile *a, int value);
/*
* check if (*a == oldvalue)
* if true: set *a to newvalue, return 1
* if false: return 0
*/
TH_API int THAtomicCompareAndSwap(int volatile *a, int oldvalue, int newvalue);
/******************************************************************************
* refcounting functions
******************************************************************************/
/*
* *a++
*/
TH_API void THAtomicIncrementRef(int volatile *a);
/*
* *a--,
* return 1 if *a == 0 after the operation, 0 otherwise
*/
TH_API int THAtomicDecrementRef(int volatile *a);
/******************************************************************************
* functions for long type
******************************************************************************/
/*
* *a = newvalue
*/
TH_API void THAtomicSetLong(long volatile *a, long newvalue);
/*
* return *a
*/
TH_API long THAtomicGetLong(long volatile *a);
/*
* *a += value,
* return previous *a
*/
TH_API long THAtomicAddLong(long volatile *a, long value);
/*
* check if (*a == oldvalue)
* if true: set *a to newvalue, return 1
* if false: return 0
*/
TH_API long THAtomicCompareAndSwapLong(long volatile *a, long oldvalue, long newvalue);
/******************************************************************************
* functions for ptrdiff_t type
******************************************************************************/
/*
* *a = newvalue
*/
TH_API void THAtomicSetPtrdiff(ptrdiff_t volatile *a, ptrdiff_t newvalue);
/*
* return *a
*/
TH_API ptrdiff_t THAtomicGetPtrdiff(ptrdiff_t volatile *a);
/*
* *a += value,
* return previous *a
*/
TH_API ptrdiff_t THAtomicAddPtrdiff(ptrdiff_t volatile *a, ptrdiff_t value);
/*
* check if (*a == oldvalue)
* if true: set *a to newvalue, return 1
* if false: return 0
*/
TH_API ptrdiff_t THAtomicCompareAndSwapPtrdiff(ptrdiff_t volatile *a, ptrdiff_t oldvalue, ptrdiff_t newvalue);
#if defined(USE_C11_ATOMICS) && defined(ATOMIC_INT_LOCK_FREE) && \
ATOMIC_INT_LOCK_FREE == 2
#define TH_ATOMIC_IPC_REFCOUNT 1
#elif defined(USE_MSC_ATOMICS) || defined(USE_GCC_ATOMICS)
#define TH_ATOMIC_IPC_REFCOUNT 1
#endif
#endif
|