diff --git a/libc/src/__support/CPP/atomic.h b/libc/src/__support/CPP/atomic.h index b0e90e32dadd2..5514062525cce 100644 --- a/libc/src/__support/CPP/atomic.h +++ b/libc/src/__support/CPP/atomic.h @@ -10,6 +10,7 @@ #define LLVM_LIBC_SRC_SUPPORT_CPP_ATOMIC_H #include "src/__support/macros/attributes.h" +#include "src/__support/macros/properties/architectures.h" #include "type_traits.h" @@ -96,7 +97,14 @@ template struct Atomic { // Issue a thread fence with the given memory ordering. LIBC_INLINE void atomic_thread_fence(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 + (void)mem_ord; + __nvvm_membar_sys(); +#else __atomic_thread_fence(int(mem_ord)); +#endif } } // namespace cpp