Fix a few includes occuring within internal namespaces

NVIDIA · Aug 4, 2021 · 6efb206 · 6efb206
1 parent 8711f32
commit 6efb206
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/libcxx/include/atomic b/libcxx/include/atomic
@@ -588,6 +588,14 @@ void atomic_signal_fence(memory_order m) noexcept;
                            __f == memory_order_acq_rel,   \
                         "memory order argument to atomic operation is invalid")
 
+#if defined(_LIBCUDACXX_HAS_MSVC_ATOMIC_IMPL)
+#  include <intrin.h>
+#endif
+
+#if !defined(_LIBCUDACXX_COMPILER_NVRTC)
+#  include <string.h>
+#endif
+
 _LIBCUDACXX_BEGIN_NAMESPACE_STD
 
 // Figure out what the underlying type for `memory_order` would be if it were

diff --git a/libcxx/include/support/atomic/atomic_base.h b/libcxx/include/support/atomic/atomic_base.h
@@ -11,8 +11,6 @@
 #ifndef _LIBCUDACXX_ATOMIC_BASE_H
 #define _LIBCUDACXX_ATOMIC_BASE_H
 
-#include <type_traits>
-
 template <typename _Tp, int _Sco>
 struct __cxx_atomic_base_impl {
   using __underlying_t = _Tp;

diff --git a/libcxx/include/support/atomic/atomic_cuda.h b/libcxx/include/support/atomic/atomic_cuda.h
@@ -11,10 +11,6 @@
 #  error "CUDA atomics are only supported for sm_60 and up on *nix and sm_70 and up on Windows."
 #endif
 
-#ifndef __CUDACC_RTC__
-#include <string.h>
-#endif // __CUDACC_RTC__
-
 #if !defined(__CLANG_ATOMIC_BOOL_LOCK_FREE) && !defined(__GCC_ATOMIC_BOOL_LOCK_FREE)
 #define ATOMIC_BOOL_LOCK_FREE      2
 #define ATOMIC_CHAR_LOCK_FREE      2

diff --git a/libcxx/include/support/atomic/atomic_msvc.h b/libcxx/include/support/atomic/atomic_msvc.h
@@ -12,8 +12,6 @@
 #error "This file is only for CL.EXE's benefit"
 #endif
 
-#include <intrin.h>
-
 #define _Compiler_barrier() _ReadWriteBarrier()
 
 #if defined(_M_ARM) || defined(_M_ARM64)