From eee06d1e8365ea1b1aa9363a3c6445745b002324 Mon Sep 17 00:00:00 2001 From: Garrett D'Amore Date: Sun, 5 Dec 2021 17:28:40 -0500 Subject: Provide atomic pointer support. This is initially used for TLS to make loading the engine pointer faster, eliminating a much more expensive lock operation. --- src/platform/windows/win_thread.c | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'src/platform/windows/win_thread.c') diff --git a/src/platform/windows/win_thread.c b/src/platform/windows/win_thread.c index 9c7c09d3..31e783a3 100644 --- a/src/platform/windows/win_thread.c +++ b/src/platform/windows/win_thread.c @@ -223,6 +223,18 @@ nni_atomic_set64(nni_atomic_u64 *v, uint64_t u) (void) InterlockedExchange64(&v->v, (LONGLONG) u); } +void * +nni_atomic_get_ptr(nni_atomic_ptr *v) +{ + return ((void *) (InterlockedExchangeAdd64(&v->v, 0))); +} + +void +nni_atomic_set_ptr(nni_atomic_ptr *v) +{ + (void) InterlockedExchange64(&v->v, (LONGLONG) (uintptr_t) v); +} + uint64_t nni_atomic_swap64(nni_atomic_u64 *v, uint64_t u) { -- cgit v1.2.3-70-g09d2