diff --git a/projects/rccl/src/include/plugin/nccl_net.h b/projects/rccl/src/include/plugin/nccl_net.h index 1b3ba23c48..ad572fdb15 100644 --- a/projects/rccl/src/include/plugin/nccl_net.h +++ b/projects/rccl/src/include/plugin/nccl_net.h @@ -11,6 +11,7 @@ #include "nccl_common.h" #include "net_device.h" #include +#include #define NCCL_NET_HANDLE_MAXSIZE 128 //Maximum value NCCL can accept for maxP2pBytes and maxCollBytes net properties