fix(sendrecv): add missing nccl-headers include

stack-info: PR: #576, branch: aws-nslick/stack/23 Signed-off-by: Nicholas Sielicki <nslick@amazon.com>
aws · Sep 6, 2024 · 82e010c · 82e010c
1 parent db0668c
commit 82e010c
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"