feat(rdma): constrain C linkage to init

aws · Sep 12, 2024 · a789535 · a789535
1 parent 4c7afdf
commit a789535
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/include/nccl_ofi_rdma.h b/include/nccl_ofi_rdma.h
@@ -5,10 +5,6 @@
 #ifndef NCCL_OFI_RDMA_H_
 #define NCCL_OFI_RDMA_H_
 
-#ifdef __cplusplus
-extern "C" {
-#endif
-
 #include <rdma/fabric.h>
 
 #include "nccl_ofi.h"
@@ -808,6 +804,11 @@ typedef struct nccl_net_ofi_rdma_device {
 #endif
 } nccl_net_ofi_rdma_device_t;
 
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
 /*
  * @brief	Initialize plugin with rdma protocol structures
  */