|
|
@@ -268,7 +268,7 @@ ge::Status ProfSetStepInfo(uint64_t index_id, uint16_t tag_id, rtStream_t stream |
|
|
return ge::FAILED; |
|
|
return ge::FAILED; |
|
|
} |
|
|
} |
|
|
auto &profiling_manager = ge::ProfilingManager::Instance(); |
|
|
auto &profiling_manager = ge::ProfilingManager::Instance(); |
|
|
profiling_manager.SetStepInfoIndex(index_id); |
|
|
|
|
|
|
|
|
//profiling_manager.SetStepInfoIndex(index_id); |
|
|
if (is_first_run && tag_id == kStepStart) { |
|
|
if (is_first_run && tag_id == kStepStart) { |
|
|
GE_CHK_STATUS_RET_NOLOG(profiling_manager.ProfileStepInfo(index_id, kModelId, tag_id, stream, device_id)); |
|
|
GE_CHK_STATUS_RET_NOLOG(profiling_manager.ProfileStepInfo(index_id, kModelId, tag_id, stream, device_id)); |
|
|
is_first_run = false; |
|
|
is_first_run = false; |
|
|
|