From c428035a7386c8ef7d87417ca25f8e849ab6d09c Mon Sep 17 00:00:00 2001
From: ywx998152 <1003254035@qq.com>
Date: Sat, 9 Jan 2021 11:27:27 +0800
Subject: [PATCH] ci test

---
 ge/graph/load/new_model_manager/task_info/hccl_task_info.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/ge/graph/load/new_model_manager/task_info/hccl_task_info.cc b/ge/graph/load/new_model_manager/task_info/hccl_task_info.cc
index 7b18a9a3..778dcd3d 100644
--- a/ge/graph/load/new_model_manager/task_info/hccl_task_info.cc
+++ b/ge/graph/load/new_model_manager/task_info/hccl_task_info.cc
@@ -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);