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 11, 2024 · 7853df2 · 7853df2
1 parent 2995611
commit 7853df2
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"