diff --git a/src/nccl_ofi_sendrecv.c b/src/nccl_ofi_sendrecv.c index 8048f97b2..598ed1d5c 100644 --- a/src/nccl_ofi_sendrecv.c +++ b/src/nccl_ofi_sendrecv.c @@ -12,6 +12,8 @@ #include #include +#include "nccl-headers/net.h" + #include "nccl_ofi.h" #if HAVE_CUDA #include "nccl_ofi_cuda.h"