p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!14653 get hccl_algo_value 

From: @zhupuxu
Reviewed-by: @jjfeing,@zhoufeng54
Signed-off-by: @jjfeing
This commit is contained in:
mindspore-ci-bot 2021-04-26 21:31:17 +08:00 committed by Gitee
parent e9221f158b 5cfb18951d
commit 8dbf601e7e
1 changed files with 19 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
mindspore/ccsrc/runtime/hccl_adapter/hccl_adapter.cc
View File

@ -28,6 +28,7 @@
static constexpr const char *kHcclOpsKernelInfoStore = "ops_kernel_info_hccl";
static constexpr const char *kHcclDeployModeEnv = "DEPLOY_MODE";
static constexpr const char *kHcclAlgoEnv = "HCCL_ALGO";
// following global var, thread safety is not guaranteed
static std::shared_ptr<ge::OpsKernelInfoStore> ops_kernel_info_store = nullptr;
static ge::OpsKernelBuilderPtr ops_kernel_builder = nullptr;
@ -41,16 +42,24 @@ static std::map<std::string, std::string> GenHcclOptions(uint32_t device_id, std
env_deploy_mode = "0";
}
return std::map<std::string, std::string>({{ge::OPTION_EXEC_IS_USEHCOM, "1"},
{ge::OPTION_EXEC_IS_USEHVD, "0"},
{ge::OPTION_EXEC_HCCL_FLAG, "1"},
{ge::OPTION_EXEC_DEVICE_ID, std::to_string(device_id)},
{ge::OPTION_EXEC_RANK_ID, rank_id.data()},
{ge::OPTION_EXEC_POD_NAME, rank_id.data()},
{ge::OPTION_EXEC_RANK_TABLE_FILE, rank_file.data()},
{ge::OPTION_GRAPH_RUN_MODE, "1"},
{ge::OPTION_EXEC_HCCL_FLAG, "1"},
{ge::OPTION_EXEC_DEPLOY_MODE, env_deploy_mode}});
std::map<std::string, std::string> default_options_map = {{ge::OPTION_EXEC_IS_USEHCOM, "1"},
{ge::OPTION_EXEC_IS_USEHVD, "0"},
{ge::OPTION_EXEC_HCCL_FLAG, "1"},
{ge::OPTION_EXEC_DEVICE_ID, std::to_string(device_id)},
{ge::OPTION_EXEC_RANK_ID, rank_id.data()},
{ge::OPTION_EXEC_POD_NAME, rank_id.data()},
{ge::OPTION_EXEC_RANK_TABLE_FILE, rank_file.data()},
{ge::OPTION_GRAPH_RUN_MODE, "1"},
{ge::OPTION_EXEC_HCCL_FLAG, "1"},
{ge::OPTION_EXEC_DEPLOY_MODE, env_deploy_mode}};
auto env_hccl_algo = common::GetEnv(kHcclAlgoEnv);
if (!env_hccl_algo.empty()) {
std::string ge_hccl_algo = "HCCL_algorithm";
default_options_map.emplace(ge_hccl_algo, env_hccl_algo);
}
return default_options_map;
}
bool InitHccl(uint32_t device_id, std::string_view rank_id, std::string_view rank_file) {