Searched refs:GetGroupRanks (Results 1 – 7 of 7) sorted by relevance
71 std::vector<int> GetGroupRanks(const std::string &group_name) { in GetGroupRanks() function72 return NCCLWrapper::instance().GetGroupRanks(group_name); in GetGroupRanks()
54 std::vector<int> GetGroupRanks(const std::string &group_name);
57 extern "C" EXPORT_WRAPPER std::vector<int> GetGroupRanks(const std::string &group_name);
119 std::vector<int> NCCLWrapper::GetGroupRanks(const std::string &group_name) { in GetGroupRanks() function in mindspore::device::gpu::NCCLWrapper
44 typedef std::vector<int> (*GetGroupRanks)(const std::string &); typedef
128 … reinterpret_cast<GetGroupRanks>(dlsym(const_cast<void *>(collective_handle_), "GetGroupRanks")); in Init()
144 … reinterpret_cast<GetGroupRanks>(dlsym(const_cast<void *>(collective_handle_), "GetGroupRanks")); in Init()