Searched refs:get_rank_size (Results 1 – 5 of 5) sorted by relevance
35 int MPIInitializer::get_rank_size(const std::string &group) { return GetGroupSize(group); } in get_rank_size() function in mindspore::device::gpu::MPIInitializer40 mpi_initializer.def("get_rank_size", &MPIInitializer::get_rank_size, "get rank size"); in PYBIND11_MODULE()
32 static int get_rank_size(const std::string &groups);
265 size = mpi.get_rank_size(group)268 size = hccl.get_rank_size()270 size = hccl.get_rank_size(group)272 size = mpi.get_rank_size(group)
144 def get_rank_size(group="hccl_world_group"): function
63 def get_rank_size(group=None): function