From 9da7bede879a22aa775740ed0070119304b21ddb Mon Sep 17 00:00:00 2001
From: weiyang <yangwei79@huawei.com>
Date: Wed, 18 Nov 2020 16:32:35 +0800
Subject: [PATCH] rts error code decoupling

---
 ge/graph/load/new_model_manager/davinci_model.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/ge/graph/load/new_model_manager/davinci_model.cc b/ge/graph/load/new_model_manager/davinci_model.cc
index d00c2eda..e6a9c589 100755
--- a/ge/graph/load/new_model_manager/davinci_model.cc
+++ b/ge/graph/load/new_model_manager/davinci_model.cc
@@ -89,6 +89,8 @@ const uint32_t kDumpL1FusionOpMByteSize = 2 * 1024 * 1024;
 const uint32_t kDumpFlagOfL1Fusion = 0;
 const char *const kDefaultBatchLable = "Batch_default";
 const int32_t kInvalidStream = -1;
+const uint32_t kEndOfSequence = 0x0704000a;
+const uint32_t kEndOfSequenceNew = 507005;
 
 inline bool IsDataOp(const std::string &node_type) {
   return node_type == DATA_TYPE || node_type == AIPP_DATA_TYPE || node_type == ANN_DATA_TYPE;
@@ -2582,7 +2584,7 @@ void *DavinciModel::Run(DavinciModel *model) {
       GE_TIMESTAMP_START(rtStreamSynchronize);
       GELOGI("rtStreamSynchronize start.");
       rt_ret = rtStreamSynchronize(model->rt_model_stream_);
-      if (rt_ret == RT_ERROR_END_OF_SEQUENCE) {
+      if (rt_ret == kEndOfSequence || rt_ret == kEndOfSequenceNew) {
         seq_end_flag = true;
       }
       GE_IF_BOOL_EXEC(