diff --git a/include/rabit/c_api.h b/include/rabit/c_api.h index 07ee08663..c833aa037 100644 --- a/include/rabit/c_api.h +++ b/include/rabit/c_api.h @@ -9,6 +9,9 @@ #ifdef __cplusplus #define RABIT_EXTERN_C extern "C" +#include +#else +#include #endif #if defined(_MSC_VER) || defined(_WIN32)