diff --git a/src/common.h b/src/common.h index 8e7cfdf027..462beba61f 100644 --- a/src/common.h +++ b/src/common.h @@ -11,6 +11,7 @@ #include "rccl/rccl.h" #include #include +#include #include #ifdef MPI_SUPPORT #include "mpi.h"