Add missing type_traits to host.h

Co-authored-by: Michael Schellenberger Costa <[email protected]>
NVIDIA · May 9, 2024 · 1cdffdb · 1cdffdb
1 parent 50e37d7
commit 1cdffdb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/libcudacxx/include/cuda/std/__atomic/functions/host.h b/libcudacxx/include/cuda/std/__atomic/functions/host.h
@@ -23,6 +23,9 @@
 
 #include <cuda/std/__atomic/order.h>
 #include <cuda/std/__atomic/platform.h>
+#include <cuda/std/__type_traits/enable_if.h>
+#include <cuda/std/__type_traits/is_floating_point.h>
+#include <cuda/std/__type_traits/remove_cvref.h>
 
 _LIBCUDACXX_BEGIN_NAMESPACE_STD