Browse Source

For step profiling.

pull/1308/head
unknown 4 years ago
parent
commit
c0f8247985
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      ge/hybrid/executor/hybrid_model_executor.cc

+ 4
- 4
ge/hybrid/executor/hybrid_model_executor.cc View File

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


Loading…
Cancel
Save