Browse Source

fix geruntime missing files and error codes

tags/v1.2.0
yanghaoran 3 years ago
parent
commit
e9e5dd7b9d
3 changed files with 6 additions and 2 deletions
  1. +3
    -0
      ge/ge_runtime/CMakeLists.txt
  2. +2
    -2
      ge/ge_runtime/runtime_model.cc
  3. +1
    -0
      ge/ge_runtime/task/task.h

+ 3
- 0
ge/ge_runtime/CMakeLists.txt View File

@@ -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})


+ 2
- 2
ge/ge_runtime/runtime_model.cc View File

@@ -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);


+ 1
- 0
ge/ge_runtime/task/task.h View File

@@ -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 {


Loading…
Cancel
Save