m53297601
/
graphengine
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!322 rts error code decoupling

Browse Source 
From: @youui
Reviewed-by: @xchu42,@liujunzhu
Signed-off-by: @liujunzhu
pull/322/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
parent 3a152ffa3f 9da7bede87
commit 88b8b33762
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File

4
ge/graph/load/new_model_manager/davinci_model.cc
Unescape View File

@ -89,6 +89,8 @@ const uint32_t kDumpL1FusionOpMByteSize = 2 * 1024 * 1024;
const uint32_t kDumpFlagOfL1Fusion = 0; const uint32_t kDumpFlagOfL1Fusion = 0;
const char *const kDefaultBatchLable = "Batch_default"; const char *const kDefaultBatchLable = "Batch_default";
const int32_t kInvalidStream = -1; const int32_t kInvalidStream = -1;
const uint32_t kEndOfSequence = 0x0704000a;
const uint32_t kEndOfSequenceNew = 507005;
inline bool IsDataOp(const std::string &node_type) { inline bool IsDataOp(const std::string &node_type) {
return node_type == DATA_TYPE || node_type == AIPP_DATA_TYPE || node_type == ANN_DATA_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); GE_TIMESTAMP_START(rtStreamSynchronize);
GELOGI("rtStreamSynchronize start."); GELOGI("rtStreamSynchronize start.");
rt_ret = rtStreamSynchronize(model->rt_model_stream_); 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; seq_end_flag = true;
} }
GE_IF_BOOL_EXEC( GE_IF_BOOL_EXEC(

Write Preview
Loading…
Cancel
Save