@@ -13,6 +13,9 @@ set(GE_SRC_LIST | |||||
"task/hccl_task.cc" | "task/hccl_task.cc" | ||||
"task/memcpy_async_task.cc" | "task/memcpy_async_task.cc" | ||||
"task/profiler_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}) | add_library(ge_runtime SHARED ${GE_SRC_LIST}) | ||||
@@ -307,8 +307,8 @@ bool RuntimeModel::Run() { | |||||
ret = rtStreamSynchronize(rt_model_stream_); | ret = rtStreamSynchronize(rt_model_stream_); | ||||
if (ret != RT_ERROR_NONE) { | 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; | return true; | ||||
} | } | ||||
GELOGE(RT_FAILED, "Model stream sync failed, ret = 0x%X", ret); | GELOGE(RT_FAILED, "Model stream sync failed, ret = 0x%X", ret); | ||||
@@ -24,6 +24,7 @@ | |||||
#include "runtime/rt_model.h" | #include "runtime/rt_model.h" | ||||
#include "ge_runtime/model_context.h" | #include "ge_runtime/model_context.h" | ||||
#include "ge_runtime/task_info.h" | #include "ge_runtime/task_info.h" | ||||
#include "external/runtime/rt_error_codes.h" | |||||
namespace ge { | namespace ge { | ||||
namespace model_runner { | namespace model_runner { | ||||