[libc] Update fence to use scoped fence now that it's supported

[jhuber6]

Summary:
Adds support for scoped fences now that the NVPTX backend doesn't break
on them.

[llvmbot]

@llvm/pr-subscribers-libc

Author: Joseph Huber (jhuber6)

Changes

Summary:
Adds support for scoped fences now that the NVPTX backend doesn't break
on them.

---

Full diff: https://github.com/llvm/llvm-project/pull/119018.diff

1 Files Affected:

• (modified) libc/src/__support/CPP/atomic.h (+7-6)

diff --git a/libc/src/__support/CPP/atomic.h b/libc/src/__support/CPP/atomic.h
index c67e4e9b6f1cbc..b4b86fd511e248 100644
--- a/libc/src/__support/CPP/atomic.h
+++ b/libc/src/__support/CPP/atomic.h
@@ -214,13 +214,14 @@ template <typename T> struct Atomic {
 };
 
 // Issue a thread fence with the given memory ordering.
-LIBC_INLINE void atomic_thread_fence([[maybe_unused]] MemoryOrder mem_ord) {
-// The NVPTX backend currently does not support atomic thread fences so we use a
-// full system fence instead.
-#ifdef LIBC_TARGET_ARCH_IS_NVPTX
-  __nvvm_membar_sys();
+LIBC_INLINE void atomic_thread_fence(
+    MemoryOrder mem_ord,
+    [[maybe_unused]] MemoryScope mem_scope = MemoryScope::DEVICE) {
+#if __has_builtin(__scoped_atomic_thread_fence)
+  return __scoped_atomic_thread_fence(static_cast<int>(mem_ord),
+                                      static_cast<int>(mem_scope));
 #else
-  __atomic_thread_fence(static_cast<int>(mem_ord));
+  return __atomic_thread_fence(static_cast<int>(mem_ord));
 #endif
 }