diff --git a/src/include/internal/host/nvshmemi_mem_transport.hpp b/src/include/internal/host/nvshmemi_mem_transport.hpp index 37a626b..bfb0580 100644 --- a/src/include/internal/host/nvshmemi_mem_transport.hpp +++ b/src/include/internal/host/nvshmemi_mem_transport.hpp @@ -8,7 +8,9 @@ #define NVSHMEMI_MEM_TRANSPORT_HPP #include -#include +#if defined(HAVE_STDBOOL_H) && (__cplusplus < 201703L) +# include +#endif #include #include #include diff --git a/src/include/internal/host/nvshmemi_symmetric_heap.hpp b/src/include/internal/host/nvshmemi_symmetric_heap.hpp index 3e61ec5..d570286 100644 --- a/src/include/internal/host/nvshmemi_symmetric_heap.hpp +++ b/src/include/internal/host/nvshmemi_symmetric_heap.hpp @@ -9,7 +9,10 @@ #include #include -#include +#if defined(HAVE_STDBOOL_H) && (__cplusplus < 201703L) +# include +#endif + #include #include #include