Searched refs:communi_parallel_mode (Results 1 – 6 of 6) sorted by relevance
25 communi_parallel_mode="same_server_group_parallel")32 communi_parallel_mode = context.get_auto_parallel_context("communi_parallel_mode")39 assert communi_parallel_mode == "same_server_group_parallel"88 context.set_auto_parallel_context(communi_parallel_mode="wrong_mode")111 communi_parallel_mode = context.get_auto_parallel_context("communi_parallel_mode")123 assert communi_parallel_mode == "all_group_parallel"
526 def set_communi_parallel_mode(self, communi_parallel_mode): argument536 if not isinstance(communi_parallel_mode, str):538 but got {type(communi_parallel_mode)}.")540 ret = self._context_handle.set_communi_parallel_mode(communi_parallel_mode)542 … raise ValueError("Communication parallel mode does not support {}".format(communi_parallel_mode))685 communi_parallel_mode=str, optimizer_weight_shard_size=int,
191 bool ParallelContext::set_communi_parallel_mode(const std::string &communi_parallel_mode) { in set_communi_parallel_mode() argument192 …:find(COMMUNI_PARALLEL_MODE_LIST.begin(), COMMUNI_PARALLEL_MODE_LIST.end(), communi_parallel_mode); in set_communi_parallel_mode()194 MS_LOG(INFO) << "Invalid communication parallel mode:" << communi_parallel_mode; in set_communi_parallel_mode()198 communi_parallel_mode_ = communi_parallel_mode; in set_communi_parallel_mode()
128 bool set_communi_parallel_mode(const std::string &communi_parallel_mode);129 std::string communi_parallel_mode() const { return communi_parallel_mode_; } in communi_parallel_mode() function
70 string communi_parallel_mode = parallel::ParallelContext::GetInstance()->communi_parallel_mode(); in DoGetHcomGroup() local71 if (communi_parallel_mode == parallel::ALL_GROUP_PARALLEL) { in DoGetHcomGroup()75 if (communi_parallel_mode == parallel::NO_GROUP_PARALLEL) { in DoGetHcomGroup()217 …Communication parallel mode: " << parallel::ParallelContext::GetInstance()->communi_parallel_mode() in AssignStream()
190 ….def("get_communi_parallel_mode", &ParallelContext::communi_parallel_mode, "Get communication para… in PYBIND11_MODULE()