未验证 提交 ffbda80c 编写于 作者: R risemeup1 提交者: GitHub

Fix test_fleet_exe_dist_model_run bug (#48492)

* fix test_fleet_exe_dist_model_run

* test
上级 428c01d6
...@@ -436,7 +436,11 @@ bool DistModel::PrepareFleetExe() { ...@@ -436,7 +436,11 @@ bool DistModel::PrepareFleetExe() {
for (int i = 0; i < config_.nranks; ++i) { for (int i = 0; i < config_.nranks; ++i) {
RankInfo *rank_info = executor_desc_.add_cluster_info(); RankInfo *rank_info = executor_desc_.add_cluster_info();
rank_info->set_rank(i); rank_info->set_rank(i);
rank_info->set_ip_port(config_.trainer_endpoints[i]); if (config_.nranks == 1) {
rank_info->set_ip_port("");
} else {
rank_info->set_ip_port(config_.trainer_endpoints[i]);
}
id_to_rank.insert({i, i}); id_to_rank.insert({i, i});
} }
fleet_exe.reset(new FleetExecutor(executor_desc_)); fleet_exe.reset(new FleetExecutor(executor_desc_));
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册