123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284 |
- /*
- * kmp_os.h -- KPTS runtime header file.
- */
- //===----------------------------------------------------------------------===//
- //
- // 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
- //
- //===----------------------------------------------------------------------===//
- #ifndef KMP_OS_H
- #define KMP_OS_H
- #include "kmp_config.h"
- #include <atomic>
- #include <stdarg.h>
- #include <stdlib.h>
- #include <string.h>
- #define KMP_FTN_PLAIN 1
- #define KMP_FTN_APPEND 2
- #define KMP_FTN_UPPER 3
- /*
- #define KMP_FTN_PREPEND 4
- #define KMP_FTN_UAPPEND 5
- */
- #define KMP_PTR_SKIP (sizeof(void *))
- /* -------------------------- Compiler variations ------------------------ */
- #define KMP_OFF 0
- #define KMP_ON 1
- #define KMP_MEM_CONS_VOLATILE 0
- #define KMP_MEM_CONS_FENCE 1
- #ifndef KMP_MEM_CONS_MODEL
- #define KMP_MEM_CONS_MODEL KMP_MEM_CONS_VOLATILE
- #endif
- #ifndef __has_cpp_attribute
- #define __has_cpp_attribute(x) 0
- #endif
- #ifndef __has_attribute
- #define __has_attribute(x) 0
- #endif
- /* ------------------------- Compiler recognition ---------------------- */
- #define KMP_COMPILER_ICC 0
- #define KMP_COMPILER_GCC 0
- #define KMP_COMPILER_CLANG 0
- #define KMP_COMPILER_MSVC 0
- #define KMP_COMPILER_ICX 0
- #if __INTEL_CLANG_COMPILER
- #undef KMP_COMPILER_ICX
- #define KMP_COMPILER_ICX 1
- #elif defined(__INTEL_COMPILER)
- #undef KMP_COMPILER_ICC
- #define KMP_COMPILER_ICC 1
- #elif defined(__clang__)
- #undef KMP_COMPILER_CLANG
- #define KMP_COMPILER_CLANG 1
- #elif defined(__GNUC__)
- #undef KMP_COMPILER_GCC
- #define KMP_COMPILER_GCC 1
- #elif defined(_MSC_VER)
- #undef KMP_COMPILER_MSVC
- #define KMP_COMPILER_MSVC 1
- #else
- #error Unknown compiler
- #endif
- #if (KMP_OS_LINUX || KMP_OS_WINDOWS || KMP_OS_FREEBSD)
- #define KMP_AFFINITY_SUPPORTED 1
- #if KMP_OS_WINDOWS && KMP_ARCH_X86_64
- #define KMP_GROUP_AFFINITY 1
- #else
- #define KMP_GROUP_AFFINITY 0
- #endif
- #else
- #define KMP_AFFINITY_SUPPORTED 0
- #define KMP_GROUP_AFFINITY 0
- #endif
- #if (KMP_OS_LINUX || (KMP_OS_FREEBSD && __FreeBSD_version >= 1301000))
- #define KMP_HAVE_SCHED_GETCPU 1
- #else
- #define KMP_HAVE_SCHED_GETCPU 0
- #endif
- /* Check for quad-precision extension. */
- #define KMP_HAVE_QUAD 0
- #if KMP_ARCH_X86 || KMP_ARCH_X86_64
- #if KMP_COMPILER_ICC || KMP_COMPILER_ICX
- /* _Quad is already defined for icc */
- #undef KMP_HAVE_QUAD
- #define KMP_HAVE_QUAD 1
- #elif KMP_COMPILER_CLANG
- /* Clang doesn't support a software-implemented
- 128-bit extended precision type yet */
- typedef long double _Quad;
- #elif KMP_COMPILER_GCC
- /* GCC on NetBSD lacks __multc3/__divtc3 builtins needed for quad */
- #if !KMP_OS_NETBSD
- typedef __float128 _Quad;
- #undef KMP_HAVE_QUAD
- #define KMP_HAVE_QUAD 1
- #endif
- #elif KMP_COMPILER_MSVC
- typedef long double _Quad;
- #endif
- #else
- #if __LDBL_MAX_EXP__ >= 16384 && KMP_COMPILER_GCC
- typedef long double _Quad;
- #undef KMP_HAVE_QUAD
- #define KMP_HAVE_QUAD 1
- #endif
- #endif /* KMP_ARCH_X86 || KMP_ARCH_X86_64 */
- #define KMP_USE_X87CONTROL 0
- #if KMP_OS_WINDOWS
- #define KMP_END_OF_LINE "\r\n"
- typedef char kmp_int8;
- typedef unsigned char kmp_uint8;
- typedef short kmp_int16;
- typedef unsigned short kmp_uint16;
- typedef int kmp_int32;
- typedef unsigned int kmp_uint32;
- #define KMP_INT32_SPEC "d"
- #define KMP_UINT32_SPEC "u"
- #ifndef KMP_STRUCT64
- typedef __int64 kmp_int64;
- typedef unsigned __int64 kmp_uint64;
- #define KMP_INT64_SPEC "I64d"
- #define KMP_UINT64_SPEC "I64u"
- #else
- struct kmp_struct64 {
- kmp_int32 a, b;
- };
- typedef struct kmp_struct64 kmp_int64;
- typedef struct kmp_struct64 kmp_uint64;
- /* Not sure what to use for KMP_[U]INT64_SPEC here */
- #endif
- #if KMP_ARCH_X86 && KMP_MSVC_COMPAT
- #undef KMP_USE_X87CONTROL
- #define KMP_USE_X87CONTROL 1
- #endif
- #if KMP_ARCH_X86_64 || KMP_ARCH_AARCH64
- #define KMP_INTPTR 1
- typedef __int64 kmp_intptr_t;
- typedef unsigned __int64 kmp_uintptr_t;
- #define KMP_INTPTR_SPEC "I64d"
- #define KMP_UINTPTR_SPEC "I64u"
- #endif
- #endif /* KMP_OS_WINDOWS */
- #if KMP_OS_UNIX
- #define KMP_END_OF_LINE "\n"
- typedef char kmp_int8;
- typedef unsigned char kmp_uint8;
- typedef short kmp_int16;
- typedef unsigned short kmp_uint16;
- typedef int kmp_int32;
- typedef unsigned int kmp_uint32;
- typedef long long kmp_int64;
- typedef unsigned long long kmp_uint64;
- #define KMP_INT32_SPEC "d"
- #define KMP_UINT32_SPEC "u"
- #define KMP_INT64_SPEC "lld"
- #define KMP_UINT64_SPEC "llu"
- #endif /* KMP_OS_UNIX */
- #if KMP_ARCH_X86 || KMP_ARCH_ARM || KMP_ARCH_MIPS
- #define KMP_SIZE_T_SPEC KMP_UINT32_SPEC
- #elif KMP_ARCH_X86_64 || KMP_ARCH_PPC64 || KMP_ARCH_AARCH64 || \
- KMP_ARCH_MIPS64 || KMP_ARCH_RISCV64
- #define KMP_SIZE_T_SPEC KMP_UINT64_SPEC
- #else
- #error "Can't determine size_t printf format specifier."
- #endif
- #if KMP_ARCH_X86
- #define KMP_SIZE_T_MAX (0xFFFFFFFF)
- #else
- #define KMP_SIZE_T_MAX (0xFFFFFFFFFFFFFFFF)
- #endif
- typedef size_t kmp_size_t;
- typedef float kmp_real32;
- typedef double kmp_real64;
- #ifndef KMP_INTPTR
- #define KMP_INTPTR 1
- typedef long kmp_intptr_t;
- typedef unsigned long kmp_uintptr_t;
- #define KMP_INTPTR_SPEC "ld"
- #define KMP_UINTPTR_SPEC "lu"
- #endif
- #ifdef BUILD_I8
- typedef kmp_int64 kmp_int;
- typedef kmp_uint64 kmp_uint;
- #else
- typedef kmp_int32 kmp_int;
- typedef kmp_uint32 kmp_uint;
- #endif /* BUILD_I8 */
- #define KMP_INT_MAX ((kmp_int32)0x7FFFFFFF)
- #define KMP_INT_MIN ((kmp_int32)0x80000000)
- // stdarg handling
- #if (KMP_ARCH_ARM || KMP_ARCH_X86_64 || KMP_ARCH_AARCH64) && \
- (KMP_OS_FREEBSD || KMP_OS_LINUX)
- typedef va_list *kmp_va_list;
- #define kmp_va_deref(ap) (*(ap))
- #define kmp_va_addr_of(ap) (&(ap))
- #else
- typedef va_list kmp_va_list;
- #define kmp_va_deref(ap) (ap)
- #define kmp_va_addr_of(ap) (ap)
- #endif
- #ifdef __cplusplus
- // macros to cast out qualifiers and to re-interpret types
- #define CCAST(type, var) const_cast<type>(var)
- #define RCAST(type, var) reinterpret_cast<type>(var)
- //-------------------------------------------------------------------------
- // template for debug prints specification ( d, u, lld, llu ), and to obtain
- // signed/unsigned flavors of a type
- template <typename T> struct traits_t {};
- // int
- template <> struct traits_t<signed int> {
- typedef signed int signed_t;
- typedef unsigned int unsigned_t;
- typedef double floating_t;
- static char const *spec;
- static const signed_t max_value = 0x7fffffff;
- static const signed_t min_value = 0x80000000;
- static const int type_size = sizeof(signed_t);
- };
- // unsigned int
- template <> struct traits_t<unsigned int> {
- typedef signed int signed_t;
- typedef unsigned int unsigned_t;
- typedef double floating_t;
- static char const *spec;
- static const unsigned_t max_value = 0xffffffff;
- static const unsigned_t min_value = 0x00000000;
- static const int type_size = sizeof(unsigned_t);
- };
- // long
- template <> struct traits_t<signed long> {
- typedef signed long signed_t;
- typedef unsigned long unsigned_t;
- typedef long double floating_t;
- static char const *spec;
- static const int type_size = sizeof(signed_t);
- };
- // long long
- template <> struct traits_t<signed long long> {
- typedef signed long long signed_t;
- typedef unsigned long long unsigned_t;
- typedef long double floating_t;
- static char const *spec;
- static const signed_t max_value = 0x7fffffffffffffffLL;
- static const signed_t min_value = 0x8000000000000000LL;
- static const int type_size = sizeof(signed_t);
- };
- // unsigned long long
- template <> struct traits_t<unsigned long long> {
- typedef signed long long signed_t;
- typedef unsigned long long unsigned_t;
- typedef long double floating_t;
- static char const *spec;
- static const unsigned_t max_value = 0xffffffffffffffffLL;
- static const unsigned_t min_value = 0x0000000000000000LL;
- static const int type_size = sizeof(unsigned_t);
- };
- //-------------------------------------------------------------------------
- #else
- #define CCAST(type, var) (type)(var)
- #define RCAST(type, var) (type)(var)
- #endif // __cplusplus
- #define KMP_EXPORT extern /* export declaration in guide libraries */
- #if __GNUC__ >= 4 && !defined(__MINGW32__)
- #define __forceinline __inline
- #endif
- /* Check if the OS/arch can support user-level mwait */
- // All mwait code tests for UMWAIT first, so it should only fall back to ring3
- // MWAIT for KNL.
- #define KMP_HAVE_MWAIT \
- ((KMP_ARCH_X86 || KMP_ARCH_X86_64) && (KMP_OS_LINUX || KMP_OS_WINDOWS) && \
- !KMP_MIC2)
- #define KMP_HAVE_UMWAIT \
- ((KMP_ARCH_X86 || KMP_ARCH_X86_64) && (KMP_OS_LINUX || KMP_OS_WINDOWS) && \
- !KMP_MIC)
- #if KMP_OS_WINDOWS
- #include <windows.h>
- static inline int KMP_GET_PAGE_SIZE(void) {
- SYSTEM_INFO si;
- GetSystemInfo(&si);
- return si.dwPageSize;
- }
- #else
- #define KMP_GET_PAGE_SIZE() getpagesize()
- #endif
- #define PAGE_ALIGNED(_addr) \
- (!((size_t)_addr & (size_t)(KMP_GET_PAGE_SIZE() - 1)))
- #define ALIGN_TO_PAGE(x) \
- (void *)(((size_t)(x)) & ~((size_t)(KMP_GET_PAGE_SIZE() - 1)))
- /* ---------- Support for cache alignment, padding, etc. ----------------*/
- #ifdef __cplusplus
- extern "C" {
- #endif // __cplusplus
- #define INTERNODE_CACHE_LINE 4096 /* for multi-node systems */
- /* Define the default size of the cache line */
- #ifndef CACHE_LINE
- #define CACHE_LINE 128 /* cache line size in bytes */
- #else
- #if (CACHE_LINE < 64) && !defined(KMP_OS_DARWIN)
- // 2006-02-13: This produces too many warnings on OS X*. Disable for now
- #warning CACHE_LINE is too small.
- #endif
- #endif /* CACHE_LINE */
- #define KMP_CACHE_PREFETCH(ADDR) /* nothing */
- // Define attribute that indicates that the fall through from the previous
- // case label is intentional and should not be diagnosed by a compiler
- // Code from libcxx/include/__config
- // Use a function like macro to imply that it must be followed by a semicolon
- #if __cplusplus > 201402L && __has_cpp_attribute(fallthrough)
- #define KMP_FALLTHROUGH() [[fallthrough]]
- // icc cannot properly tell this attribute is absent so force off
- #elif KMP_COMPILER_ICC
- #define KMP_FALLTHROUGH() ((void)0)
- #elif __has_cpp_attribute(clang::fallthrough)
- #define KMP_FALLTHROUGH() [[clang::fallthrough]]
- #elif __has_attribute(fallthrough) || __GNUC__ >= 7
- #define KMP_FALLTHROUGH() __attribute__((__fallthrough__))
- #else
- #define KMP_FALLTHROUGH() ((void)0)
- #endif
- #if KMP_HAVE_ATTRIBUTE_WAITPKG
- #define KMP_ATTRIBUTE_TARGET_WAITPKG __attribute__((target("waitpkg")))
- #else
- #define KMP_ATTRIBUTE_TARGET_WAITPKG /* Nothing */
- #endif
- #if KMP_HAVE_ATTRIBUTE_RTM
- #define KMP_ATTRIBUTE_TARGET_RTM __attribute__((target("rtm")))
- #else
- #define KMP_ATTRIBUTE_TARGET_RTM /* Nothing */
- #endif
- // Define attribute that indicates a function does not return
- #if __cplusplus >= 201103L
- #define KMP_NORETURN [[noreturn]]
- #elif KMP_OS_WINDOWS
- #define KMP_NORETURN __declspec(noreturn)
- #else
- #define KMP_NORETURN __attribute__((noreturn))
- #endif
- #if KMP_OS_WINDOWS && KMP_MSVC_COMPAT
- #define KMP_ALIGN(bytes) __declspec(align(bytes))
- #define KMP_THREAD_LOCAL __declspec(thread)
- #define KMP_ALIAS /* Nothing */
- #else
- #define KMP_ALIGN(bytes) __attribute__((aligned(bytes)))
- #define KMP_THREAD_LOCAL __thread
- #define KMP_ALIAS(alias_of) __attribute__((alias(alias_of)))
- #endif
- #if KMP_HAVE_WEAK_ATTRIBUTE && !KMP_DYNAMIC_LIB
- #define KMP_WEAK_ATTRIBUTE_EXTERNAL __attribute__((weak))
- #else
- #define KMP_WEAK_ATTRIBUTE_EXTERNAL /* Nothing */
- #endif
- #if KMP_HAVE_WEAK_ATTRIBUTE
- #define KMP_WEAK_ATTRIBUTE_INTERNAL __attribute__((weak))
- #else
- #define KMP_WEAK_ATTRIBUTE_INTERNAL /* Nothing */
- #endif
- // Define KMP_VERSION_SYMBOL and KMP_EXPAND_NAME
- #ifndef KMP_STR
- #define KMP_STR(x) _KMP_STR(x)
- #define _KMP_STR(x) #x
- #endif
- #ifdef KMP_USE_VERSION_SYMBOLS
- // If using versioned symbols, KMP_EXPAND_NAME prepends
- // __kmp_api_ to the real API name
- #define KMP_EXPAND_NAME(api_name) _KMP_EXPAND_NAME(api_name)
- #define _KMP_EXPAND_NAME(api_name) __kmp_api_##api_name
- #define KMP_VERSION_SYMBOL(api_name, ver_num, ver_str) \
- _KMP_VERSION_SYMBOL(api_name, ver_num, ver_str, "VERSION")
- #define _KMP_VERSION_SYMBOL(api_name, ver_num, ver_str, default_ver) \
- __typeof__(__kmp_api_##api_name) __kmp_api_##api_name##_##ver_num##_alias \
- __attribute__((alias(KMP_STR(__kmp_api_##api_name)))); \
- __asm__( \
- ".symver " KMP_STR(__kmp_api_##api_name##_##ver_num##_alias) "," KMP_STR( \
- api_name) "@" ver_str "\n\t"); \
- __asm__(".symver " KMP_STR(__kmp_api_##api_name) "," KMP_STR( \
- api_name) "@@" default_ver "\n\t")
- #define KMP_VERSION_OMPC_SYMBOL(apic_name, api_name, ver_num, ver_str) \
- _KMP_VERSION_OMPC_SYMBOL(apic_name, api_name, ver_num, ver_str, "VERSION")
- #define _KMP_VERSION_OMPC_SYMBOL(apic_name, api_name, ver_num, ver_str, \
- default_ver) \
- __typeof__(__kmp_api_##apic_name) __kmp_api_##apic_name##_##ver_num##_alias \
- __attribute__((alias(KMP_STR(__kmp_api_##apic_name)))); \
- __asm__(".symver " KMP_STR(__kmp_api_##apic_name) "," KMP_STR( \
- apic_name) "@@" default_ver "\n\t"); \
- __asm__( \
- ".symver " KMP_STR(__kmp_api_##apic_name##_##ver_num##_alias) "," KMP_STR( \
- api_name) "@" ver_str "\n\t")
- #else // KMP_USE_VERSION_SYMBOLS
- #define KMP_EXPAND_NAME(api_name) api_name
- #define KMP_VERSION_SYMBOL(api_name, ver_num, ver_str) /* Nothing */
- #define KMP_VERSION_OMPC_SYMBOL(apic_name, api_name, ver_num, \
- ver_str) /* Nothing */
- #endif // KMP_USE_VERSION_SYMBOLS
- /* Temporary note: if performance testing of this passes, we can remove
- all references to KMP_DO_ALIGN and replace with KMP_ALIGN. */
- #define KMP_DO_ALIGN(bytes) KMP_ALIGN(bytes)
- #define KMP_ALIGN_CACHE KMP_ALIGN(CACHE_LINE)
- #define KMP_ALIGN_CACHE_INTERNODE KMP_ALIGN(INTERNODE_CACHE_LINE)
- /* General purpose fence types for memory operations */
- enum kmp_mem_fence_type {
- kmp_no_fence, /* No memory fence */
- kmp_acquire_fence, /* Acquire (read) memory fence */
- kmp_release_fence, /* Release (write) memory fence */
- kmp_full_fence /* Full (read+write) memory fence */
- };
- // Synchronization primitives
- #if KMP_ASM_INTRINS && KMP_OS_WINDOWS
- #if KMP_MSVC_COMPAT && !KMP_COMPILER_CLANG
- #pragma intrinsic(InterlockedExchangeAdd)
- #pragma intrinsic(InterlockedCompareExchange)
- #pragma intrinsic(InterlockedExchange)
- #if !(KMP_COMPILER_ICX && KMP_32_BIT_ARCH)
- #pragma intrinsic(InterlockedExchange64)
- #endif
- #endif
- // Using InterlockedIncrement / InterlockedDecrement causes a library loading
- // ordering problem, so we use InterlockedExchangeAdd instead.
- #define KMP_TEST_THEN_INC32(p) InterlockedExchangeAdd((volatile long *)(p), 1)
- #define KMP_TEST_THEN_INC_ACQ32(p) \
- InterlockedExchangeAdd((volatile long *)(p), 1)
- #define KMP_TEST_THEN_ADD4_32(p) InterlockedExchangeAdd((volatile long *)(p), 4)
- #define KMP_TEST_THEN_ADD4_ACQ32(p) \
- InterlockedExchangeAdd((volatile long *)(p), 4)
- #define KMP_TEST_THEN_DEC32(p) InterlockedExchangeAdd((volatile long *)(p), -1)
- #define KMP_TEST_THEN_DEC_ACQ32(p) \
- InterlockedExchangeAdd((volatile long *)(p), -1)
- #define KMP_TEST_THEN_ADD32(p, v) \
- InterlockedExchangeAdd((volatile long *)(p), (v))
- #define KMP_COMPARE_AND_STORE_RET32(p, cv, sv) \
- InterlockedCompareExchange((volatile long *)(p), (long)(sv), (long)(cv))
- #define KMP_XCHG_FIXED32(p, v) \
- InterlockedExchange((volatile long *)(p), (long)(v))
- #define KMP_XCHG_FIXED64(p, v) \
- InterlockedExchange64((volatile kmp_int64 *)(p), (kmp_int64)(v))
- inline kmp_real32 KMP_XCHG_REAL32(volatile kmp_real32 *p, kmp_real32 v) {
- kmp_int32 tmp = InterlockedExchange((volatile long *)p, *(long *)&v);
- return *(kmp_real32 *)&tmp;
- }
- #define KMP_TEST_THEN_OR8(p, v) __kmp_test_then_or8((p), (v))
- #define KMP_TEST_THEN_AND8(p, v) __kmp_test_then_and8((p), (v))
- #define KMP_TEST_THEN_OR32(p, v) __kmp_test_then_or32((p), (v))
- #define KMP_TEST_THEN_AND32(p, v) __kmp_test_then_and32((p), (v))
- #define KMP_TEST_THEN_OR64(p, v) __kmp_test_then_or64((p), (v))
- #define KMP_TEST_THEN_AND64(p, v) __kmp_test_then_and64((p), (v))
- extern kmp_int8 __kmp_test_then_or8(volatile kmp_int8 *p, kmp_int8 v);
- extern kmp_int8 __kmp_test_then_and8(volatile kmp_int8 *p, kmp_int8 v);
- extern kmp_int32 __kmp_test_then_add32(volatile kmp_int32 *p, kmp_int32 v);
- extern kmp_uint32 __kmp_test_then_or32(volatile kmp_uint32 *p, kmp_uint32 v);
- extern kmp_uint32 __kmp_test_then_and32(volatile kmp_uint32 *p, kmp_uint32 v);
- extern kmp_int64 __kmp_test_then_add64(volatile kmp_int64 *p, kmp_int64 v);
- extern kmp_uint64 __kmp_test_then_or64(volatile kmp_uint64 *p, kmp_uint64 v);
- extern kmp_uint64 __kmp_test_then_and64(volatile kmp_uint64 *p, kmp_uint64 v);
- #if KMP_ARCH_AARCH64 && KMP_COMPILER_MSVC && !KMP_COMPILER_CLANG
- #define KMP_TEST_THEN_INC64(p) _InterlockedExchangeAdd64((p), 1LL)
- #define KMP_TEST_THEN_INC_ACQ64(p) _InterlockedExchangeAdd64_acq((p), 1LL)
- #define KMP_TEST_THEN_ADD4_64(p) _InterlockedExchangeAdd64((p), 4LL)
- // #define KMP_TEST_THEN_ADD4_ACQ64(p) _InterlockedExchangeAdd64_acq((p), 4LL)
- // #define KMP_TEST_THEN_DEC64(p) _InterlockedExchangeAdd64((p), -1LL)
- // #define KMP_TEST_THEN_DEC_ACQ64(p) _InterlockedExchangeAdd64_acq((p), -1LL)
- // #define KMP_TEST_THEN_ADD8(p, v) _InterlockedExchangeAdd8((p), (v))
- #define KMP_TEST_THEN_ADD64(p, v) _InterlockedExchangeAdd64((p), (v))
- #define KMP_COMPARE_AND_STORE_ACQ8(p, cv, sv) \
- __kmp_compare_and_store_acq8((p), (cv), (sv))
- #define KMP_COMPARE_AND_STORE_REL8(p, cv, sv) \
- __kmp_compare_and_store_rel8((p), (cv), (sv))
- #define KMP_COMPARE_AND_STORE_ACQ16(p, cv, sv) \
- __kmp_compare_and_store_acq16((p), (cv), (sv))
- /*
- #define KMP_COMPARE_AND_STORE_REL16(p, cv, sv) \
- __kmp_compare_and_store_rel16((p), (cv), (sv))
- */
- #define KMP_COMPARE_AND_STORE_ACQ32(p, cv, sv) \
- __kmp_compare_and_store_acq32((volatile kmp_int32 *)(p), (kmp_int32)(cv), \
- (kmp_int32)(sv))
- #define KMP_COMPARE_AND_STORE_REL32(p, cv, sv) \
- __kmp_compare_and_store_rel32((volatile kmp_int32 *)(p), (kmp_int32)(cv), \
- (kmp_int32)(sv))
- #define KMP_COMPARE_AND_STORE_ACQ64(p, cv, sv) \
- __kmp_compare_and_store_acq64((volatile kmp_int64 *)(p), (kmp_int64)(cv), \
- (kmp_int64)(sv))
- #define KMP_COMPARE_AND_STORE_REL64(p, cv, sv) \
- __kmp_compare_and_store_rel64((volatile kmp_int64 *)(p), (kmp_int64)(cv), \
- (kmp_int64)(sv))
- #define KMP_COMPARE_AND_STORE_PTR(p, cv, sv) \
- __kmp_compare_and_store_ptr((void *volatile *)(p), (void *)(cv), (void *)(sv))
- // KMP_COMPARE_AND_STORE expects this order: pointer, compare, exchange
- // _InterlockedCompareExchange expects this order: pointer, exchange, compare
- // KMP_COMPARE_AND_STORE also returns a bool indicating a successful write. A
- // write is successful if the return value of _InterlockedCompareExchange is the
- // same as the compare value.
- inline kmp_int8 __kmp_compare_and_store_acq8(volatile kmp_int8 *p, kmp_int8 cv,
- kmp_int8 sv) {
- return _InterlockedCompareExchange8_acq(p, sv, cv) == cv;
- }
- inline kmp_int8 __kmp_compare_and_store_rel8(volatile kmp_int8 *p, kmp_int8 cv,
- kmp_int8 sv) {
- return _InterlockedCompareExchange8_rel(p, sv, cv) == cv;
- }
- inline kmp_int16 __kmp_compare_and_store_acq16(volatile kmp_int16 *p,
- kmp_int16 cv, kmp_int16 sv) {
- return _InterlockedCompareExchange16_acq(p, sv, cv) == cv;
- }
- inline kmp_int16 __kmp_compare_and_store_rel16(volatile kmp_int16 *p,
- kmp_int16 cv, kmp_int16 sv) {
- return _InterlockedCompareExchange16_rel(p, sv, cv) == cv;
- }
- inline kmp_int32 __kmp_compare_and_store_acq32(volatile kmp_int32 *p,
- kmp_int32 cv, kmp_int32 sv) {
- return _InterlockedCompareExchange_acq((volatile long *)p, sv, cv) == cv;
- }
- inline kmp_int32 __kmp_compare_and_store_rel32(volatile kmp_int32 *p,
- kmp_int32 cv, kmp_int32 sv) {
- return _InterlockedCompareExchange_rel((volatile long *)p, sv, cv) == cv;
- }
- inline kmp_int32 __kmp_compare_and_store_acq64(volatile kmp_int64 *p,
- kmp_int64 cv, kmp_int64 sv) {
- return _InterlockedCompareExchange64_acq(p, sv, cv) == cv;
- }
- inline kmp_int32 __kmp_compare_and_store_rel64(volatile kmp_int64 *p,
- kmp_int64 cv, kmp_int64 sv) {
- return _InterlockedCompareExchange64_rel(p, sv, cv) == cv;
- }
- inline kmp_int32 __kmp_compare_and_store_ptr(void *volatile *p, void *cv,
- void *sv) {
- return _InterlockedCompareExchangePointer(p, sv, cv) == cv;
- }
- // The _RET versions return the value instead of a bool
- /*
- #define KMP_COMPARE_AND_STORE_RET8(p, cv, sv) \
- _InterlockedCompareExchange8((p), (sv), (cv))
- #define KMP_COMPARE_AND_STORE_RET16(p, cv, sv) \
- _InterlockedCompareExchange16((p), (sv), (cv))
- */
- #define KMP_COMPARE_AND_STORE_RET64(p, cv, sv) \
- _InterlockedCompareExchange64((volatile kmp_int64 *)(p), (kmp_int64)(sv), \
- (kmp_int64)(cv))
- /*
- #define KMP_XCHG_FIXED8(p, v) \
- _InterlockedExchange8((volatile kmp_int8 *)(p), (kmp_int8)(v));
- */
- // #define KMP_XCHG_FIXED16(p, v) _InterlockedExchange16((p), (v));
- // #define KMP_XCHG_REAL64(p, v) __kmp_xchg_real64((p), (v)));
- // inline kmp_real64 __kmp_xchg_real64(volatile kmp_real64 *p, kmp_real64 v) {
- // kmp_int64 tmp = _InterlockedExchange64((volatile kmp_int64 *)p, *(kmp_int64
- // *)&v); return *(kmp_real64 *)&tmp;
- // }
- #else // !KMP_ARCH_AARCH64
- // Routines that we still need to implement in assembly.
- extern kmp_int8 __kmp_test_then_add8(volatile kmp_int8 *p, kmp_int8 v);
- extern kmp_int8 __kmp_compare_and_store8(volatile kmp_int8 *p, kmp_int8 cv,
- kmp_int8 sv);
- extern kmp_int16 __kmp_compare_and_store16(volatile kmp_int16 *p, kmp_int16 cv,
- kmp_int16 sv);
- extern kmp_int32 __kmp_compare_and_store32(volatile kmp_int32 *p, kmp_int32 cv,
- kmp_int32 sv);
- extern kmp_int32 __kmp_compare_and_store64(volatile kmp_int64 *p, kmp_int64 cv,
- kmp_int64 sv);
- extern kmp_int8 __kmp_compare_and_store_ret8(volatile kmp_int8 *p, kmp_int8 cv,
- kmp_int8 sv);
- extern kmp_int16 __kmp_compare_and_store_ret16(volatile kmp_int16 *p,
- kmp_int16 cv, kmp_int16 sv);
- extern kmp_int32 __kmp_compare_and_store_ret32(volatile kmp_int32 *p,
- kmp_int32 cv, kmp_int32 sv);
- extern kmp_int64 __kmp_compare_and_store_ret64(volatile kmp_int64 *p,
- kmp_int64 cv, kmp_int64 sv);
- extern kmp_int8 __kmp_xchg_fixed8(volatile kmp_int8 *p, kmp_int8 v);
- extern kmp_int16 __kmp_xchg_fixed16(volatile kmp_int16 *p, kmp_int16 v);
- extern kmp_int32 __kmp_xchg_fixed32(volatile kmp_int32 *p, kmp_int32 v);
- extern kmp_int64 __kmp_xchg_fixed64(volatile kmp_int64 *p, kmp_int64 v);
- extern kmp_real32 __kmp_xchg_real32(volatile kmp_real32 *p, kmp_real32 v);
- extern kmp_real64 __kmp_xchg_real64(volatile kmp_real64 *p, kmp_real64 v);
- //#define KMP_TEST_THEN_INC32(p) __kmp_test_then_add32((p), 1)
- //#define KMP_TEST_THEN_INC_ACQ32(p) __kmp_test_then_add32((p), 1)
- #define KMP_TEST_THEN_INC64(p) __kmp_test_then_add64((p), 1LL)
- #define KMP_TEST_THEN_INC_ACQ64(p) __kmp_test_then_add64((p), 1LL)
- //#define KMP_TEST_THEN_ADD4_32(p) __kmp_test_then_add32((p), 4)
- //#define KMP_TEST_THEN_ADD4_ACQ32(p) __kmp_test_then_add32((p), 4)
- #define KMP_TEST_THEN_ADD4_64(p) __kmp_test_then_add64((p), 4LL)
- #define KMP_TEST_THEN_ADD4_ACQ64(p) __kmp_test_then_add64((p), 4LL)
- //#define KMP_TEST_THEN_DEC32(p) __kmp_test_then_add32((p), -1)
- //#define KMP_TEST_THEN_DEC_ACQ32(p) __kmp_test_then_add32((p), -1)
- #define KMP_TEST_THEN_DEC64(p) __kmp_test_then_add64((p), -1LL)
- #define KMP_TEST_THEN_DEC_ACQ64(p) __kmp_test_then_add64((p), -1LL)
- //#define KMP_TEST_THEN_ADD32(p, v) __kmp_test_then_add32((p), (v))
- #define KMP_TEST_THEN_ADD8(p, v) __kmp_test_then_add8((p), (v))
- #define KMP_TEST_THEN_ADD64(p, v) __kmp_test_then_add64((p), (v))
- #define KMP_COMPARE_AND_STORE_ACQ8(p, cv, sv) \
- __kmp_compare_and_store8((p), (cv), (sv))
- #define KMP_COMPARE_AND_STORE_REL8(p, cv, sv) \
- __kmp_compare_and_store8((p), (cv), (sv))
- #define KMP_COMPARE_AND_STORE_ACQ16(p, cv, sv) \
- __kmp_compare_and_store16((p), (cv), (sv))
- #define KMP_COMPARE_AND_STORE_REL16(p, cv, sv) \
- __kmp_compare_and_store16((p), (cv), (sv))
- #define KMP_COMPARE_AND_STORE_ACQ32(p, cv, sv) \
- __kmp_compare_and_store32((volatile kmp_int32 *)(p), (kmp_int32)(cv), \
- (kmp_int32)(sv))
- #define KMP_COMPARE_AND_STORE_REL32(p, cv, sv) \
- __kmp_compare_and_store32((volatile kmp_int32 *)(p), (kmp_int32)(cv), \
- (kmp_int32)(sv))
- #define KMP_COMPARE_AND_STORE_ACQ64(p, cv, sv) \
- __kmp_compare_and_store64((volatile kmp_int64 *)(p), (kmp_int64)(cv), \
- (kmp_int64)(sv))
- #define KMP_COMPARE_AND_STORE_REL64(p, cv, sv) \
- __kmp_compare_and_store64((volatile kmp_int64 *)(p), (kmp_int64)(cv), \
- (kmp_int64)(sv))
- #if KMP_ARCH_X86
- #define KMP_COMPARE_AND_STORE_PTR(p, cv, sv) \
- __kmp_compare_and_store32((volatile kmp_int32 *)(p), (kmp_int32)(cv), \
- (kmp_int32)(sv))
- #else /* 64 bit pointers */
- #define KMP_COMPARE_AND_STORE_PTR(p, cv, sv) \
- __kmp_compare_and_store64((volatile kmp_int64 *)(p), (kmp_int64)(cv), \
- (kmp_int64)(sv))
- #endif /* KMP_ARCH_X86 */
- #define KMP_COMPARE_AND_STORE_RET8(p, cv, sv) \
- __kmp_compare_and_store_ret8((p), (cv), (sv))
- #define KMP_COMPARE_AND_STORE_RET16(p, cv, sv) \
- __kmp_compare_and_store_ret16((p), (cv), (sv))
- #define KMP_COMPARE_AND_STORE_RET64(p, cv, sv) \
- __kmp_compare_and_store_ret64((volatile kmp_int64 *)(p), (kmp_int64)(cv), \
- (kmp_int64)(sv))
- #define KMP_XCHG_FIXED8(p, v) \
- __kmp_xchg_fixed8((volatile kmp_int8 *)(p), (kmp_int8)(v));
- #define KMP_XCHG_FIXED16(p, v) __kmp_xchg_fixed16((p), (v));
- //#define KMP_XCHG_FIXED32(p, v) __kmp_xchg_fixed32((p), (v));
- //#define KMP_XCHG_FIXED64(p, v) __kmp_xchg_fixed64((p), (v));
- //#define KMP_XCHG_REAL32(p, v) __kmp_xchg_real32((p), (v));
- #define KMP_XCHG_REAL64(p, v) __kmp_xchg_real64((p), (v));
- #endif
- #elif (KMP_ASM_INTRINS && KMP_OS_UNIX) || !(KMP_ARCH_X86 || KMP_ARCH_X86_64)
- /* cast p to correct type so that proper intrinsic will be used */
- #define KMP_TEST_THEN_INC32(p) \
- __sync_fetch_and_add((volatile kmp_int32 *)(p), 1)
- #define KMP_TEST_THEN_INC_ACQ32(p) \
- __sync_fetch_and_add((volatile kmp_int32 *)(p), 1)
- #if KMP_ARCH_MIPS
- #define KMP_TEST_THEN_INC64(p) \
- __atomic_fetch_add((volatile kmp_int64 *)(p), 1LL, __ATOMIC_SEQ_CST)
- #define KMP_TEST_THEN_INC_ACQ64(p) \
- __atomic_fetch_add((volatile kmp_int64 *)(p), 1LL, __ATOMIC_SEQ_CST)
- #else
- #define KMP_TEST_THEN_INC64(p) \
- __sync_fetch_and_add((volatile kmp_int64 *)(p), 1LL)
- #define KMP_TEST_THEN_INC_ACQ64(p) \
- __sync_fetch_and_add((volatile kmp_int64 *)(p), 1LL)
- #endif
- #define KMP_TEST_THEN_ADD4_32(p) \
- __sync_fetch_and_add((volatile kmp_int32 *)(p), 4)
- #define KMP_TEST_THEN_ADD4_ACQ32(p) \
- __sync_fetch_and_add((volatile kmp_int32 *)(p), 4)
- #if KMP_ARCH_MIPS
- #define KMP_TEST_THEN_ADD4_64(p) \
- __atomic_fetch_add((volatile kmp_int64 *)(p), 4LL, __ATOMIC_SEQ_CST)
- #define KMP_TEST_THEN_ADD4_ACQ64(p) \
- __atomic_fetch_add((volatile kmp_int64 *)(p), 4LL, __ATOMIC_SEQ_CST)
- #define KMP_TEST_THEN_DEC64(p) \
- __atomic_fetch_sub((volatile kmp_int64 *)(p), 1LL, __ATOMIC_SEQ_CST)
- #define KMP_TEST_THEN_DEC_ACQ64(p) \
- __atomic_fetch_sub((volatile kmp_int64 *)(p), 1LL, __ATOMIC_SEQ_CST)
- #else
- #define KMP_TEST_THEN_ADD4_64(p) \
- __sync_fetch_and_add((volatile kmp_int64 *)(p), 4LL)
- #define KMP_TEST_THEN_ADD4_ACQ64(p) \
- __sync_fetch_and_add((volatile kmp_int64 *)(p), 4LL)
- #define KMP_TEST_THEN_DEC64(p) \
- __sync_fetch_and_sub((volatile kmp_int64 *)(p), 1LL)
- #define KMP_TEST_THEN_DEC_ACQ64(p) \
- __sync_fetch_and_sub((volatile kmp_int64 *)(p), 1LL)
- #endif
- #define KMP_TEST_THEN_DEC32(p) \
- __sync_fetch_and_sub((volatile kmp_int32 *)(p), 1)
- #define KMP_TEST_THEN_DEC_ACQ32(p) \
- __sync_fetch_and_sub((volatile kmp_int32 *)(p), 1)
- #define KMP_TEST_THEN_ADD8(p, v) \
- __sync_fetch_and_add((volatile kmp_int8 *)(p), (kmp_int8)(v))
- #define KMP_TEST_THEN_ADD32(p, v) \
- __sync_fetch_and_add((volatile kmp_int32 *)(p), (kmp_int32)(v))
- #if KMP_ARCH_MIPS
- #define KMP_TEST_THEN_ADD64(p, v) \
- __atomic_fetch_add((volatile kmp_uint64 *)(p), (kmp_uint64)(v), \
- __ATOMIC_SEQ_CST)
- #else
- #define KMP_TEST_THEN_ADD64(p, v) \
- __sync_fetch_and_add((volatile kmp_int64 *)(p), (kmp_int64)(v))
- #endif
- #define KMP_TEST_THEN_OR8(p, v) \
- __sync_fetch_and_or((volatile kmp_int8 *)(p), (kmp_int8)(v))
- #define KMP_TEST_THEN_AND8(p, v) \
- __sync_fetch_and_and((volatile kmp_int8 *)(p), (kmp_int8)(v))
- #define KMP_TEST_THEN_OR32(p, v) \
- __sync_fetch_and_or((volatile kmp_uint32 *)(p), (kmp_uint32)(v))
- #define KMP_TEST_THEN_AND32(p, v) \
- __sync_fetch_and_and((volatile kmp_uint32 *)(p), (kmp_uint32)(v))
- #if KMP_ARCH_MIPS
- #define KMP_TEST_THEN_OR64(p, v) \
- __atomic_fetch_or((volatile kmp_uint64 *)(p), (kmp_uint64)(v), \
- __ATOMIC_SEQ_CST)
- #define KMP_TEST_THEN_AND64(p, v) \
- __atomic_fetch_and((volatile kmp_uint64 *)(p), (kmp_uint64)(v), \
- __ATOMIC_SEQ_CST)
- #else
- #define KMP_TEST_THEN_OR64(p, v) \
- __sync_fetch_and_or((volatile kmp_uint64 *)(p), (kmp_uint64)(v))
- #define KMP_TEST_THEN_AND64(p, v) \
- __sync_fetch_and_and((volatile kmp_uint64 *)(p), (kmp_uint64)(v))
- #endif
- #define KMP_COMPARE_AND_STORE_ACQ8(p, cv, sv) \
- __sync_bool_compare_and_swap((volatile kmp_uint8 *)(p), (kmp_uint8)(cv), \
- (kmp_uint8)(sv))
- #define KMP_COMPARE_AND_STORE_REL8(p, cv, sv) \
- __sync_bool_compare_and_swap((volatile kmp_uint8 *)(p), (kmp_uint8)(cv), \
- (kmp_uint8)(sv))
- #define KMP_COMPARE_AND_STORE_ACQ16(p, cv, sv) \
- __sync_bool_compare_and_swap((volatile kmp_uint16 *)(p), (kmp_uint16)(cv), \
- (kmp_uint16)(sv))
- #define KMP_COMPARE_AND_STORE_REL16(p, cv, sv) \
- __sync_bool_compare_and_swap((volatile kmp_uint16 *)(p), (kmp_uint16)(cv), \
- (kmp_uint16)(sv))
- #define KMP_COMPARE_AND_STORE_ACQ32(p, cv, sv) \
- __sync_bool_compare_and_swap((volatile kmp_uint32 *)(p), (kmp_uint32)(cv), \
- (kmp_uint32)(sv))
- #define KMP_COMPARE_AND_STORE_REL32(p, cv, sv) \
- __sync_bool_compare_and_swap((volatile kmp_uint32 *)(p), (kmp_uint32)(cv), \
- (kmp_uint32)(sv))
- #define KMP_COMPARE_AND_STORE_PTR(p, cv, sv) \
- __sync_bool_compare_and_swap((void *volatile *)(p), (void *)(cv), \
- (void *)(sv))
- #define KMP_COMPARE_AND_STORE_RET8(p, cv, sv) \
- __sync_val_compare_and_swap((volatile kmp_uint8 *)(p), (kmp_uint8)(cv), \
- (kmp_uint8)(sv))
- #define KMP_COMPARE_AND_STORE_RET16(p, cv, sv) \
- __sync_val_compare_and_swap((volatile kmp_uint16 *)(p), (kmp_uint16)(cv), \
- (kmp_uint16)(sv))
- #define KMP_COMPARE_AND_STORE_RET32(p, cv, sv) \
- __sync_val_compare_and_swap((volatile kmp_uint32 *)(p), (kmp_uint32)(cv), \
- (kmp_uint32)(sv))
- #if KMP_ARCH_MIPS
- static inline bool mips_sync_bool_compare_and_swap(volatile kmp_uint64 *p,
- kmp_uint64 cv,
- kmp_uint64 sv) {
- return __atomic_compare_exchange(p, &cv, &sv, false, __ATOMIC_SEQ_CST,
- __ATOMIC_SEQ_CST);
- }
- static inline bool mips_sync_val_compare_and_swap(volatile kmp_uint64 *p,
- kmp_uint64 cv,
- kmp_uint64 sv) {
- __atomic_compare_exchange(p, &cv, &sv, false, __ATOMIC_SEQ_CST,
- __ATOMIC_SEQ_CST);
- return cv;
- }
- #define KMP_COMPARE_AND_STORE_ACQ64(p, cv, sv) \
- mips_sync_bool_compare_and_swap((volatile kmp_uint64 *)(p), \
- (kmp_uint64)(cv), (kmp_uint64)(sv))
- #define KMP_COMPARE_AND_STORE_REL64(p, cv, sv) \
- mips_sync_bool_compare_and_swap((volatile kmp_uint64 *)(p), \
- (kmp_uint64)(cv), (kmp_uint64)(sv))
- #define KMP_COMPARE_AND_STORE_RET64(p, cv, sv) \
- mips_sync_val_compare_and_swap((volatile kmp_uint64 *)(p), (kmp_uint64)(cv), \
- (kmp_uint64)(sv))
- #else
- #define KMP_COMPARE_AND_STORE_ACQ64(p, cv, sv) \
- __sync_bool_compare_and_swap((volatile kmp_uint64 *)(p), (kmp_uint64)(cv), \
- (kmp_uint64)(sv))
- #define KMP_COMPARE_AND_STORE_REL64(p, cv, sv) \
- __sync_bool_compare_and_swap((volatile kmp_uint64 *)(p), (kmp_uint64)(cv), \
- (kmp_uint64)(sv))
- #define KMP_COMPARE_AND_STORE_RET64(p, cv, sv) \
- __sync_val_compare_and_swap((volatile kmp_uint64 *)(p), (kmp_uint64)(cv), \
- (kmp_uint64)(sv))
- #endif
- #if KMP_OS_DARWIN && defined(__INTEL_COMPILER) && __INTEL_COMPILER >= 1800
- #define KMP_XCHG_FIXED8(p, v) \
- __atomic_exchange_1((volatile kmp_uint8 *)(p), (kmp_uint8)(v), \
- __ATOMIC_SEQ_CST)
- #else
- #define KMP_XCHG_FIXED8(p, v) \
- __sync_lock_test_and_set((volatile kmp_uint8 *)(p), (kmp_uint8)(v))
- #endif
- #define KMP_XCHG_FIXED16(p, v) \
- __sync_lock_test_and_set((volatile kmp_uint16 *)(p), (kmp_uint16)(v))
- #define KMP_XCHG_FIXED32(p, v) \
- __sync_lock_test_and_set((volatile kmp_uint32 *)(p), (kmp_uint32)(v))
- #define KMP_XCHG_FIXED64(p, v) \
- __sync_lock_test_and_set((volatile kmp_uint64 *)(p), (kmp_uint64)(v))
- inline kmp_real32 KMP_XCHG_REAL32(volatile kmp_real32 *p, kmp_real32 v) {
- volatile kmp_uint32 *up;
- kmp_uint32 uv;
- memcpy(&up, &p, sizeof(up));
- memcpy(&uv, &v, sizeof(uv));
- kmp_int32 tmp = __sync_lock_test_and_set(up, uv);
- kmp_real32 ftmp;
- memcpy(&ftmp, &tmp, sizeof(tmp));
- return ftmp;
- }
- inline kmp_real64 KMP_XCHG_REAL64(volatile kmp_real64 *p, kmp_real64 v) {
- volatile kmp_uint64 *up;
- kmp_uint64 uv;
- memcpy(&up, &p, sizeof(up));
- memcpy(&uv, &v, sizeof(uv));
- kmp_int64 tmp = __sync_lock_test_and_set(up, uv);
- kmp_real64 dtmp;
- memcpy(&dtmp, &tmp, sizeof(tmp));
- return dtmp;
- }
- #else
- extern kmp_int8 __kmp_test_then_add8(volatile kmp_int8 *p, kmp_int8 v);
- extern kmp_int8 __kmp_test_then_or8(volatile kmp_int8 *p, kmp_int8 v);
- extern kmp_int8 __kmp_test_then_and8(volatile kmp_int8 *p, kmp_int8 v);
- extern kmp_int32 __kmp_test_then_add32(volatile kmp_int32 *p, kmp_int32 v);
- extern kmp_uint32 __kmp_test_then_or32(volatile kmp_uint32 *p, kmp_uint32 v);
- extern kmp_uint32 __kmp_test_then_and32(volatile kmp_uint32 *p, kmp_uint32 v);
- extern kmp_int64 __kmp_test_then_add64(volatile kmp_int64 *p, kmp_int64 v);
- extern kmp_uint64 __kmp_test_then_or64(volatile kmp_uint64 *p, kmp_uint64 v);
- extern kmp_uint64 __kmp_test_then_and64(volatile kmp_uint64 *p, kmp_uint64 v);
- extern kmp_int8 __kmp_compare_and_store8(volatile kmp_int8 *p, kmp_int8 cv,
- kmp_int8 sv);
- extern kmp_int16 __kmp_compare_and_store16(volatile kmp_int16 *p, kmp_int16 cv,
- kmp_int16 sv);
- extern kmp_int32 __kmp_compare_and_store32(volatile kmp_int32 *p, kmp_int32 cv,
- kmp_int32 sv);
- extern kmp_int32 __kmp_compare_and_store64(volatile kmp_int64 *p, kmp_int64 cv,
- kmp_int64 sv);
- extern kmp_int8 __kmp_compare_and_store_ret8(volatile kmp_int8 *p, kmp_int8 cv,
- kmp_int8 sv);
- extern kmp_int16 __kmp_compare_and_store_ret16(volatile kmp_int16 *p,
- kmp_int16 cv, kmp_int16 sv);
- extern kmp_int32 __kmp_compare_and_store_ret32(volatile kmp_int32 *p,
- kmp_int32 cv, kmp_int32 sv);
- extern kmp_int64 __kmp_compare_and_store_ret64(volatile kmp_int64 *p,
- kmp_int64 cv, kmp_int64 sv);
- extern kmp_int8 __kmp_xchg_fixed8(volatile kmp_int8 *p, kmp_int8 v);
- extern kmp_int16 __kmp_xchg_fixed16(volatile kmp_int16 *p, kmp_int16 v);
- extern kmp_int32 __kmp_xchg_fixed32(volatile kmp_int32 *p, kmp_int32 v);
- extern kmp_int64 __kmp_xchg_fixed64(volatile kmp_int64 *p, kmp_int64 v);
- extern kmp_real32 __kmp_xchg_real32(volatile kmp_real32 *p, kmp_real32 v);
- extern kmp_real64 __kmp_xchg_real64(volatile kmp_real64 *p, kmp_real64 v);
- #define KMP_TEST_THEN_INC32(p) \
- __kmp_test_then_add32((volatile kmp_int32 *)(p), 1)
- #define KMP_TEST_THEN_INC_ACQ32(p) \
- __kmp_test_then_add32((volatile kmp_int32 *)(p), 1)
- #define KMP_TEST_THEN_INC64(p) \
- __kmp_test_then_add64((volatile kmp_int64 *)(p), 1LL)
- #define KMP_TEST_THEN_INC_ACQ64(p) \
- __kmp_test_then_add64((volatile kmp_int64 *)(p), 1LL)
- #define KMP_TEST_THEN_ADD4_32(p) \
- __kmp_test_then_add32((volatile kmp_int32 *)(p), 4)
- #define KMP_TEST_THEN_ADD4_ACQ32(p) \
- __kmp_test_then_add32((volatile kmp_int32 *)(p), 4)
- #define KMP_TEST_THEN_ADD4_64(p) \
- __kmp_test_then_add64((volatile kmp_int64 *)(p), 4LL)
- #define KMP_TEST_THEN_ADD4_ACQ64(p) \
- __kmp_test_then_add64((volatile kmp_int64 *)(p), 4LL)
- #define KMP_TEST_THEN_DEC32(p) \
- __kmp_test_then_add32((volatile kmp_int32 *)(p), -1)
- #define KMP_TEST_THEN_DEC_ACQ32(p) \
- __kmp_test_then_add32((volatile kmp_int32 *)(p), -1)
- #define KMP_TEST_THEN_DEC64(p) \
- __kmp_test_then_add64((volatile kmp_int64 *)(p), -1LL)
- #define KMP_TEST_THEN_DEC_ACQ64(p) \
- __kmp_test_then_add64((volatile kmp_int64 *)(p), -1LL)
- #define KMP_TEST_THEN_ADD8(p, v) \
- __kmp_test_then_add8((volatile kmp_int8 *)(p), (kmp_int8)(v))
- #define KMP_TEST_THEN_ADD32(p, v) \
- __kmp_test_then_add32((volatile kmp_int32 *)(p), (kmp_int32)(v))
- #define KMP_TEST_THEN_ADD64(p, v) \
- __kmp_test_then_add64((volatile kmp_int64 *)(p), (kmp_int64)(v))
- #define KMP_TEST_THEN_OR8(p, v) \
- __kmp_test_then_or8((volatile kmp_int8 *)(p), (kmp_int8)(v))
- #define KMP_TEST_THEN_AND8(p, v) \
- __kmp_test_then_and8((volatile kmp_int8 *)(p), (kmp_int8)(v))
- #define KMP_TEST_THEN_OR32(p, v) \
- __kmp_test_then_or32((volatile kmp_uint32 *)(p), (kmp_uint32)(v))
- #define KMP_TEST_THEN_AND32(p, v) \
- __kmp_test_then_and32((volatile kmp_uint32 *)(p), (kmp_uint32)(v))
- #define KMP_TEST_THEN_OR64(p, v) \
- __kmp_test_then_or64((volatile kmp_uint64 *)(p), (kmp_uint64)(v))
- #define KMP_TEST_THEN_AND64(p, v) \
- __kmp_test_then_and64((volatile kmp_uint64 *)(p), (kmp_uint64)(v))
- #define KMP_COMPARE_AND_STORE_ACQ8(p, cv, sv) \
- __kmp_compare_and_store8((volatile kmp_int8 *)(p), (kmp_int8)(cv), \
- (kmp_int8)(sv))
- #define KMP_COMPARE_AND_STORE_REL8(p, cv, sv) \
- __kmp_compare_and_store8((volatile kmp_int8 *)(p), (kmp_int8)(cv), \
- (kmp_int8)(sv))
- #define KMP_COMPARE_AND_STORE_ACQ16(p, cv, sv) \
- __kmp_compare_and_store16((volatile kmp_int16 *)(p), (kmp_int16)(cv), \
- (kmp_int16)(sv))
- #define KMP_COMPARE_AND_STORE_REL16(p, cv, sv) \
- __kmp_compare_and_store16((volatile kmp_int16 *)(p), (kmp_int16)(cv), \
- (kmp_int16)(sv))
- #define KMP_COMPARE_AND_STORE_ACQ32(p, cv, sv) \
- __kmp_compare_and_store32((volatile kmp_int32 *)(p), (kmp_int32)(cv), \
- (kmp_int32)(sv))
- #define KMP_COMPARE_AND_STORE_REL32(p, cv, sv) \
- __kmp_compare_and_store32((volatile kmp_int32 *)(p), (kmp_int32)(cv), \
- (kmp_int32)(sv))
- #define KMP_COMPARE_AND_STORE_ACQ64(p, cv, sv) \
- __kmp_compare_and_store64((volatile kmp_int64 *)(p), (kmp_int64)(cv), \
- (kmp_int64)(sv))
- #define KMP_COMPARE_AND_STORE_REL64(p, cv, sv) \
- __kmp_compare_and_store64((volatile kmp_int64 *)(p), (kmp_int64)(cv), \
- (kmp_int64)(sv))
- #if KMP_ARCH_X86
- #define KMP_COMPARE_AND_STORE_PTR(p, cv, sv) \
- __kmp_compare_and_store32((volatile kmp_int32 *)(p), (kmp_int32)(cv), \
- (kmp_int32)(sv))
- #else /* 64 bit pointers */
- #define KMP_COMPARE_AND_STORE_PTR(p, cv, sv) \
- __kmp_compare_and_store64((volatile kmp_int64 *)(p), (kmp_int64)(cv), \
- (kmp_int64)(sv))
- #endif /* KMP_ARCH_X86 */
- #define KMP_COMPARE_AND_STORE_RET8(p, cv, sv) \
- __kmp_compare_and_store_ret8((p), (cv), (sv))
- #define KMP_COMPARE_AND_STORE_RET16(p, cv, sv) \
- __kmp_compare_and_store_ret16((p), (cv), (sv))
- #define KMP_COMPARE_AND_STORE_RET32(p, cv, sv) \
- __kmp_compare_and_store_ret32((volatile kmp_int32 *)(p), (kmp_int32)(cv), \
- (kmp_int32)(sv))
- #define KMP_COMPARE_AND_STORE_RET64(p, cv, sv) \
- __kmp_compare_and_store_ret64((volatile kmp_int64 *)(p), (kmp_int64)(cv), \
- (kmp_int64)(sv))
- #define KMP_XCHG_FIXED8(p, v) \
- __kmp_xchg_fixed8((volatile kmp_int8 *)(p), (kmp_int8)(v));
- #define KMP_XCHG_FIXED16(p, v) __kmp_xchg_fixed16((p), (v));
- #define KMP_XCHG_FIXED32(p, v) __kmp_xchg_fixed32((p), (v));
- #define KMP_XCHG_FIXED64(p, v) __kmp_xchg_fixed64((p), (v));
- #define KMP_XCHG_REAL32(p, v) __kmp_xchg_real32((p), (v));
- #define KMP_XCHG_REAL64(p, v) __kmp_xchg_real64((p), (v));
- #endif /* KMP_ASM_INTRINS */
- /* ------------- relaxed consistency memory model stuff ------------------ */
- #if KMP_OS_WINDOWS
- #ifdef __ABSOFT_WIN
- #define KMP_MB() asm("nop")
- #define KMP_IMB() asm("nop")
- #else
- #define KMP_MB() /* _asm{ nop } */
- #define KMP_IMB() /* _asm{ nop } */
- #endif
- #endif /* KMP_OS_WINDOWS */
- #if KMP_ARCH_PPC64 || KMP_ARCH_ARM || KMP_ARCH_AARCH64 || KMP_ARCH_MIPS || \
- KMP_ARCH_MIPS64 || KMP_ARCH_RISCV64
- #if KMP_OS_WINDOWS
- #undef KMP_MB
- #define KMP_MB() std::atomic_thread_fence(std::memory_order_seq_cst)
- #else /* !KMP_OS_WINDOWS */
- #define KMP_MB() __sync_synchronize()
- #endif
- #endif
- #ifndef KMP_MB
- #define KMP_MB() /* nothing to do */
- #endif
- #if KMP_ARCH_X86 || KMP_ARCH_X86_64
- #if KMP_COMPILER_ICC || KMP_COMPILER_ICX
- #define KMP_MFENCE_() _mm_mfence()
- #define KMP_SFENCE_() _mm_sfence()
- #elif KMP_COMPILER_MSVC
- #define KMP_MFENCE_() MemoryBarrier()
- #define KMP_SFENCE_() MemoryBarrier()
- #else
- #define KMP_MFENCE_() __sync_synchronize()
- #define KMP_SFENCE_() __sync_synchronize()
- #endif
- #define KMP_MFENCE() \
- if (UNLIKELY(!__kmp_cpuinfo.initialized)) { \
- __kmp_query_cpuid(&__kmp_cpuinfo); \
- } \
- if (__kmp_cpuinfo.flags.sse2) { \
- KMP_MFENCE_(); \
- }
- #define KMP_SFENCE() KMP_SFENCE_()
- #else
- #define KMP_MFENCE() KMP_MB()
- #define KMP_SFENCE() KMP_MB()
- #endif
- #ifndef KMP_IMB
- #define KMP_IMB() /* nothing to do */
- #endif
- #ifndef KMP_ST_REL32
- #define KMP_ST_REL32(A, D) (*(A) = (D))
- #endif
- #ifndef KMP_ST_REL64
- #define KMP_ST_REL64(A, D) (*(A) = (D))
- #endif
- #ifndef KMP_LD_ACQ32
- #define KMP_LD_ACQ32(A) (*(A))
- #endif
- #ifndef KMP_LD_ACQ64
- #define KMP_LD_ACQ64(A) (*(A))
- #endif
- /* ------------------------------------------------------------------------ */
- // FIXME - maybe this should this be
- //
- // #define TCR_4(a) (*(volatile kmp_int32 *)(&a))
- // #define TCW_4(a,b) (a) = (*(volatile kmp_int32 *)&(b))
- //
- // #define TCR_8(a) (*(volatile kmp_int64 *)(a))
- // #define TCW_8(a,b) (a) = (*(volatile kmp_int64 *)(&b))
- //
- // I'm fairly certain this is the correct thing to do, but I'm afraid
- // of performance regressions.
- #define TCR_1(a) (a)
- #define TCW_1(a, b) (a) = (b)
- #define TCR_4(a) (a)
- #define TCW_4(a, b) (a) = (b)
- #define TCI_4(a) (++(a))
- #define TCD_4(a) (--(a))
- #define TCR_8(a) (a)
- #define TCW_8(a, b) (a) = (b)
- #define TCI_8(a) (++(a))
- #define TCD_8(a) (--(a))
- #define TCR_SYNC_4(a) (a)
- #define TCW_SYNC_4(a, b) (a) = (b)
- #define TCX_SYNC_4(a, b, c) \
- KMP_COMPARE_AND_STORE_REL32((volatile kmp_int32 *)(volatile void *)&(a), \
- (kmp_int32)(b), (kmp_int32)(c))
- #define TCR_SYNC_8(a) (a)
- #define TCW_SYNC_8(a, b) (a) = (b)
- #define TCX_SYNC_8(a, b, c) \
- KMP_COMPARE_AND_STORE_REL64((volatile kmp_int64 *)(volatile void *)&(a), \
- (kmp_int64)(b), (kmp_int64)(c))
- #if KMP_ARCH_X86 || KMP_ARCH_MIPS
- // What about ARM?
- #define TCR_PTR(a) ((void *)TCR_4(a))
- #define TCW_PTR(a, b) TCW_4((a), (b))
- #define TCR_SYNC_PTR(a) ((void *)TCR_SYNC_4(a))
- #define TCW_SYNC_PTR(a, b) TCW_SYNC_4((a), (b))
- #define TCX_SYNC_PTR(a, b, c) ((void *)TCX_SYNC_4((a), (b), (c)))
- #else /* 64 bit pointers */
- #define TCR_PTR(a) ((void *)TCR_8(a))
- #define TCW_PTR(a, b) TCW_8((a), (b))
- #define TCR_SYNC_PTR(a) ((void *)TCR_SYNC_8(a))
- #define TCW_SYNC_PTR(a, b) TCW_SYNC_8((a), (b))
- #define TCX_SYNC_PTR(a, b, c) ((void *)TCX_SYNC_8((a), (b), (c)))
- #endif /* KMP_ARCH_X86 */
- /* If these FTN_{TRUE,FALSE} values change, may need to change several places
- where they are used to check that language is Fortran, not C. */
- #ifndef FTN_TRUE
- #define FTN_TRUE TRUE
- #endif
- #ifndef FTN_FALSE
- #define FTN_FALSE FALSE
- #endif
- typedef void (*microtask_t)(int *gtid, int *npr, ...);
- #ifdef USE_VOLATILE_CAST
- #define VOLATILE_CAST(x) (volatile x)
- #else
- #define VOLATILE_CAST(x) (x)
- #endif
- #define KMP_WAIT __kmp_wait_4
- #define KMP_WAIT_PTR __kmp_wait_4_ptr
- #define KMP_EQ __kmp_eq_4
- #define KMP_NEQ __kmp_neq_4
- #define KMP_LT __kmp_lt_4
- #define KMP_GE __kmp_ge_4
- #define KMP_LE __kmp_le_4
- /* Workaround for Intel(R) 64 code gen bug when taking address of static array
- * (Intel(R) 64 Tracker #138) */
- #if (KMP_ARCH_X86_64 || KMP_ARCH_PPC64) && KMP_OS_LINUX
- #define STATIC_EFI2_WORKAROUND
- #else
- #define STATIC_EFI2_WORKAROUND static
- #endif
- // Support of BGET usage
- #ifndef KMP_USE_BGET
- #define KMP_USE_BGET 1
- #endif
- // Switches for OSS builds
- #ifndef USE_CMPXCHG_FIX
- #define USE_CMPXCHG_FIX 1
- #endif
- // Enable dynamic user lock
- #define KMP_USE_DYNAMIC_LOCK 1
- // Enable Intel(R) Transactional Synchronization Extensions (Intel(R) TSX) if
- // dynamic user lock is turned on
- #if KMP_USE_DYNAMIC_LOCK
- // Visual studio can't handle the asm sections in this code
- #define KMP_USE_TSX (KMP_ARCH_X86 || KMP_ARCH_X86_64) && !KMP_COMPILER_MSVC
- #ifdef KMP_USE_ADAPTIVE_LOCKS
- #undef KMP_USE_ADAPTIVE_LOCKS
- #endif
- #define KMP_USE_ADAPTIVE_LOCKS KMP_USE_TSX
- #endif
- // Enable tick time conversion of ticks to seconds
- #if KMP_STATS_ENABLED
- #define KMP_HAVE_TICK_TIME \
- (KMP_OS_LINUX && (KMP_MIC || KMP_ARCH_X86 || KMP_ARCH_X86_64))
- #endif
- // Warning levels
- enum kmp_warnings_level {
- kmp_warnings_off = 0, /* No warnings */
- kmp_warnings_low, /* Minimal warnings (default) */
- kmp_warnings_explicit = 6, /* Explicitly set to ON - more warnings */
- kmp_warnings_verbose /* reserved */
- };
- #ifdef __cplusplus
- } // extern "C"
- #endif // __cplusplus
- // Safe C API
- #include "kmp_safe_c_api.h"
- // Macros for C++11 atomic functions
- #define KMP_ATOMIC_LD(p, order) (p)->load(std::memory_order_##order)
- #define KMP_ATOMIC_OP(op, p, v, order) (p)->op(v, std::memory_order_##order)
- // For non-default load/store
- #define KMP_ATOMIC_LD_ACQ(p) KMP_ATOMIC_LD(p, acquire)
- #define KMP_ATOMIC_LD_RLX(p) KMP_ATOMIC_LD(p, relaxed)
- #define KMP_ATOMIC_ST_REL(p, v) KMP_ATOMIC_OP(store, p, v, release)
- #define KMP_ATOMIC_ST_RLX(p, v) KMP_ATOMIC_OP(store, p, v, relaxed)
- // For non-default fetch_<op>
- #define KMP_ATOMIC_ADD(p, v) KMP_ATOMIC_OP(fetch_add, p, v, acq_rel)
- #define KMP_ATOMIC_SUB(p, v) KMP_ATOMIC_OP(fetch_sub, p, v, acq_rel)
- #define KMP_ATOMIC_AND(p, v) KMP_ATOMIC_OP(fetch_and, p, v, acq_rel)
- #define KMP_ATOMIC_OR(p, v) KMP_ATOMIC_OP(fetch_or, p, v, acq_rel)
- #define KMP_ATOMIC_INC(p) KMP_ATOMIC_OP(fetch_add, p, 1, acq_rel)
- #define KMP_ATOMIC_DEC(p) KMP_ATOMIC_OP(fetch_sub, p, 1, acq_rel)
- #define KMP_ATOMIC_ADD_RLX(p, v) KMP_ATOMIC_OP(fetch_add, p, v, relaxed)
- #define KMP_ATOMIC_INC_RLX(p) KMP_ATOMIC_OP(fetch_add, p, 1, relaxed)
- // Callers of the following functions cannot see the side effect on "expected".
- template <typename T>
- bool __kmp_atomic_compare_store(std::atomic<T> *p, T expected, T desired) {
- return p->compare_exchange_strong(
- expected, desired, std::memory_order_acq_rel, std::memory_order_relaxed);
- }
- template <typename T>
- bool __kmp_atomic_compare_store_acq(std::atomic<T> *p, T expected, T desired) {
- return p->compare_exchange_strong(
- expected, desired, std::memory_order_acquire, std::memory_order_relaxed);
- }
- template <typename T>
- bool __kmp_atomic_compare_store_rel(std::atomic<T> *p, T expected, T desired) {
- return p->compare_exchange_strong(
- expected, desired, std::memory_order_release, std::memory_order_relaxed);
- }
- // Symbol lookup on Linux/Windows
- #if KMP_OS_WINDOWS
- extern void *__kmp_lookup_symbol(const char *name);
- #define KMP_DLSYM(name) __kmp_lookup_symbol(name)
- #define KMP_DLSYM_NEXT(name) nullptr
- #else
- #define KMP_DLSYM(name) dlsym(RTLD_DEFAULT, name)
- #define KMP_DLSYM_NEXT(name) dlsym(RTLD_NEXT, name)
- #endif
- #endif /* KMP_OS_H */
|