Home
last modified time | relevance | path

Searched refs:GetGroupRanks (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/distribution/
Dcollective_wrapper.cc71 std::vector<int> GetGroupRanks(const std::string &group_name) { in GetGroupRanks() function
72 return NCCLWrapper::instance().GetGroupRanks(group_name); in GetGroupRanks()
Dnccl_wrapper.h54 std::vector<int> GetGroupRanks(const std::string &group_name);
Dcollective_wrapper.h57 extern "C" EXPORT_WRAPPER std::vector<int> GetGroupRanks(const std::string &group_name);
Dnccl_wrapper.cc119 std::vector<int> NCCLWrapper::GetGroupRanks(const std::string &group_name) { in GetGroupRanks() function in mindspore::device::gpu::NCCLWrapper
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/
Dnccl_gpu_kernel.h44 typedef std::vector<int> (*GetGroupRanks)(const std::string &); typedef
Dsync_batch_norm_grad_gpu_kernel.h128 … reinterpret_cast<GetGroupRanks>(dlsym(const_cast<void *>(collective_handle_), "GetGroupRanks")); in Init()
Dsync_batch_norm_gpu_kernel.h144 … reinterpret_cast<GetGroupRanks>(dlsym(const_cast<void *>(collective_handle_), "GetGroupRanks")); in Init()