From d7319181984e62c43bacbb382e86a4d053ea245d Mon Sep 17 00:00:00 2001
From: yanghaoran <yanghaoran2@huawei.com>
Date: Mon, 7 Dec 2020 17:14:14 +0800
Subject: [PATCH] fix geruntime missing files and error codes

---
 ge/ge_runtime/CMakeLists.txt   | 3 +++
 ge/ge_runtime/runtime_model.cc | 4 ++--
 ge/ge_runtime/task/task.h      | 1 +
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/ge/ge_runtime/CMakeLists.txt b/ge/ge_runtime/CMakeLists.txt
index 42d3b344..ce1b89ea 100644
--- a/ge/ge_runtime/CMakeLists.txt
+++ b/ge/ge_runtime/CMakeLists.txt
@@ -13,6 +13,9 @@ set(GE_SRC_LIST
     "task/hccl_task.cc"
     "task/memcpy_async_task.cc"
     "task/profiler_task.cc"
+    "task/label_goto_task.cc"
+    "task/label_set_task.cc"
+    "task/label_switch_task.cc"
 )
 
 add_library(ge_runtime SHARED ${GE_SRC_LIST})
diff --git a/ge/ge_runtime/runtime_model.cc b/ge/ge_runtime/runtime_model.cc
index 0ff56ef1..fb0f3e85 100644
--- a/ge/ge_runtime/runtime_model.cc
+++ b/ge/ge_runtime/runtime_model.cc
@@ -307,8 +307,8 @@ bool RuntimeModel::Run() {
 
   ret = rtStreamSynchronize(rt_model_stream_);
   if (ret != RT_ERROR_NONE) {
-    if (ret == RT_ERROR_END_OF_SEQUENCE) {
-      GELOGI("Model stream RT_ERROR_END_OF_SEQUENCE signal received, ret = 0x%X", ret);
+    if (ret == ACL_ERROR_RT_END_OF_SEQUENCE) {
+      GELOGI("Model stream ACL_ERROR_RT_END_OF_SEQUENCE signal received, ret = 0x%X", ret);
       return true;
     }
     GELOGE(RT_FAILED, "Model stream sync failed, ret = 0x%X", ret);
diff --git a/ge/ge_runtime/task/task.h b/ge/ge_runtime/task/task.h
index 6c4df248..c255fd22 100644
--- a/ge/ge_runtime/task/task.h
+++ b/ge/ge_runtime/task/task.h
@@ -24,6 +24,7 @@
 #include "runtime/rt_model.h"
 #include "ge_runtime/model_context.h"
 #include "ge_runtime/task_info.h"
+#include "external/runtime/rt_error_codes.h"
 
 namespace ge {
 namespace model_runner {