From c76789a5c961a5fa843f0d7a9135b06fa40490f3 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Wed, 15 Jul 2020 11:40:22 +0800 Subject: [PATCH] feat(mge/imperative): add io remote wrapper GitOrigin-RevId: 6dae8287d3dda50a52c3f1de86bae7a72daea397 --- src/opr-mm/impl/io_remote.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/opr-mm/impl/io_remote.cpp b/src/opr-mm/impl/io_remote.cpp index 938f816ab..49e3a5b27 100644 --- a/src/opr-mm/impl/io_remote.cpp +++ b/src/opr-mm/impl/io_remote.cpp @@ -59,7 +59,7 @@ void RemoteSend::scn_do_execute() { comp_node.get_uid()); m_megray_comm = MegRayCommBuilder::get_megray_comm( - reg_info.hash, m_key, 2, 0, MegRay::MEGRAY_UCX, m_group_client); + reg_info.hash, m_key, 2, 0, MegRay::MEGRAY_NCCL, m_group_client); m_megray_ctx = MegRay::CudaContext::make(get_stream(output(0))); @@ -158,7 +158,7 @@ void RemoteRecv::scn_do_execute() { comp_node.get_uid()); m_megray_comm = MegRayCommBuilder::get_megray_comm( - reg_info.hash, m_key, 2, 1, MegRay::MEGRAY_UCX, m_group_client); + reg_info.hash, m_key, 2, 1, MegRay::MEGRAY_NCCL, m_group_client); m_megray_ctx = MegRay::CudaContext::make(get_stream(output(0))); -- GitLab