add ncclSend and ncclRecv (#27621)

* include ncclRecv and ncclSend, test=develop
revert-27356-init_low_level_gloo
lilong12 5 years ago committed by GitHub
parent fa73e4a284
commit 5218b7af6b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -64,6 +64,13 @@ NCCL_RAND_ROUTINE_EACH(DECLARE_DYNAMIC_LOAD_NCCL_WRAP)
NCCL_RAND_ROUTINE_EACH_AFTER_2212(DECLARE_DYNAMIC_LOAD_NCCL_WRAP)
#endif
#if NCCL_VERSION_CODE >= 2703
#define NCCL_RAND_ROUTINE_EACH_AFTER_2703(__macro) \
__macro(ncclSend); \
__macro(ncclRecv);
NCCL_RAND_ROUTINE_EACH_AFTER_2703(DECLARE_DYNAMIC_LOAD_NCCL_WRAP)
#endif
} // namespace dynload
} // namespace platform
} // namespace paddle

Loading…
Cancel
Save