From c0f8247985404638e1ec06e0fa354b4fd2ad9dc4 Mon Sep 17 00:00:00 2001 From: unknown Date: Sat, 20 Mar 2021 10:58:47 +0800 Subject: [PATCH] For step profiling. --- ge/hybrid/executor/hybrid_model_executor.cc | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/ge/hybrid/executor/hybrid_model_executor.cc b/ge/hybrid/executor/hybrid_model_executor.cc index ad295abf..3a729139 100755 --- a/ge/hybrid/executor/hybrid_model_executor.cc +++ b/ge/hybrid/executor/hybrid_model_executor.cc @@ -28,7 +28,7 @@ const int kIntBase = 10; const char *const kEnvProfilingLevel = "HYBRID_PROFILING_LEVEL"; const char *const kIndexId = "index_id"; const char *const kModeleId = "model_id"; -const char *const kTimeStamp = "time_stamp"; +const char *const kTimeStamp = "time_stamp"; const char *const kStreamId = "stream_id"; const char *const kTaskId = "task_id"; const char *const kTagId = "tag_id"; @@ -93,7 +93,7 @@ Status HybridModelExecutor::ProfileStepInfo(uint16_t tag_id) { GELOGE(RT_FAILED, "[Call][rtProfilerTraceEx] failed, ret: 0x%X", rt_ret); return RT_ERROR_TO_GE_STATUS(rt_ret); } - GELOGD("Profiling Step Info TraceTask execute async success. index_id = %lu, model_id = %u, tag_id = %u", + GELOGD("Profiling Step Info TraceTask execute async success. index_id = %lu, model_id = %lu, tag_id = %u", index_id, model_id, tag_id); mmTimespec timespec = mmGetTickCount(); @@ -139,11 +139,11 @@ Status HybridModelExecutor::ExecuteGraphInternal(SubgraphExecutor &executor, RECORD_MODEL_EXECUTION_EVENT(&context_, "[InitContext] End"); // tag_id 0 means step begin, 1 meas step end. - GE_CHK_STATUS_RET_NOLOG(ProfileStepBegin(0)); + GE_CHK_STATUS_RET_NOLOG(ProfileStepInfo(0)); HYBRID_CHK_STATUS_RET(executor.ExecuteAsync(args.inputs, args.input_desc, args.outputs), "Failed to execute partitioned call."); RECORD_MODEL_EXECUTION_EVENT(&context_, "[ExecuteAsync] End"); - GE_CHK_STATUS_RET_NOLOG(ProfileStepEnd(1)); + GE_CHK_STATUS_RET_NOLOG(ProfileStepInfo(1)); HYBRID_CHK_STATUS_RET(executor.Synchronize(), "Failed to sync root graph."); RECORD_MODEL_EXECUTION_EVENT(&context_, "[Synchronize] End");