feat(rdma): constrain C linkage to init

stack-info: PR: #591, branch: aws-nslick/stack/32
aws · Sep 14, 2024 · 68750ac · 68750ac
1 parent 03c4994
commit 68750ac
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
  */