Browse Source

Pre Merge pull request !912 from yangxinmian/development

pull/912/MERGE
yangxinmian Gitee 5 years ago
parent
commit
6d84a5fac9
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      ge/graph/load/model_manager/task_info/hccl_task_info.cc

+ 4
- 0
ge/graph/load/model_manager/task_info/hccl_task_info.cc View File

@@ -319,6 +319,10 @@ void HcclTaskInfo::GetPrivateDefByTaskDef(const domi::TaskDef &task) {
GELOGE(RT_FAILED, "Call rtMallocHost Fail, ret = 0x%X.", ret);
return;
}
if (ret != 1) {
GELOGE(RT_FAILED, "Call rtMallocHost Fail, ret = 0x%X.", ret);
return;
}

ret = rtMemcpy(private_def_, private_def_len_, task.private_def().c_str(), private_def_len_,
RT_MEMCPY_HOST_TO_HOST);


Loading…
Cancel
Save