fix(sendrecv): add missing nccl-headers include

stack-info: PR: aws#576, branch: aws-nslick/stack/23 Signed-off-by: Nicholas Sielicki <nslick@amazon.com>
aws-nslick · Sep 22, 2024 · 9d34ec0 · 9d34ec0
1 parent 56c5593
commit 9d34ec0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/nccl_ofi_sendrecv.c b/src/nccl_ofi_sendrecv.c
@@ -12,6 +12,8 @@
 #include <sys/mman.h>
 #include <unistd.h>
 
+#include "nccl-headers/net.h"
+
 #include "nccl_ofi.h"
 #if HAVE_CUDA
 #include "nccl_ofi_cuda.h"