Browse Source

For step profiling.

pull/1308/head
unknown 4 years ago
parent
commit
8288f6e512
4 changed files with 4 additions and 4 deletions
  1. +1
    -1
      ge/hybrid/executor/hybrid_model_executor.cc
  2. +1
    -1
      tests/depends/runtime/src/runtime_stub.cc
  3. +1
    -1
      tests/ut/ge/hybrid/ge_hybrid_unittest.cc
  4. +1
    -1
      third_party/fwkacllib/inc/runtime/base.h

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

@@ -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 = %lu, 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();


+ 1
- 1
tests/depends/runtime/src/runtime_stub.cc View File

@@ -311,7 +311,7 @@ rtError_t rtStreamCreateWithFlags(rtStream_t *stream, int32_t priority, uint32_t

rtError_t rtFlushCache(uint64_t base, uint32_t len) { return RT_ERROR_NONE; }

rtError_t rtProfilerTrace(uint64_t id, bool notify, uint32_t flags, rtStream_t stream_) { return RT_ERROR_NONE; }
rtError_t rtProfilerTrace(uint64_t id, bool notify, uint32_t flags, rtStream_t stream) { return RT_ERROR_NONE; }

rtError_t rtProfilerTraceEx(uint64_t id, uint64_t modelId, uint16_t tagId, rtStream_t stream_) { return RT_ERROR_NONE; }



+ 1
- 1
tests/ut/ge/hybrid/ge_hybrid_unittest.cc View File

@@ -269,5 +269,5 @@ TEST_F(UtestGeHybrid, Step_info_profile) {
auto root_model = make_shared<GeRootModel>(graph);
HybridModel model(root_model);
HybridModelExecutor executor(&model, 0, nullptr);
ASSERT_EQ(executor.ProfileStepInfo(0), SUCCESS):
EXPECT_EQ(executor.ProfileStepInfo(0), SUCCESS):
}

+ 1
- 1
third_party/fwkacllib/inc/runtime/base.h View File

@@ -158,7 +158,7 @@ RTS_API rtError_t rtProfilerTrace(uint64_t id, bool notify, uint32_t flags, rtSt
* @ingroup profiling_base
* @brief ts send keypoint profiler log.
*/
RTS_API rtError_t rtProfilerTraceEx(uint64_t id, uint64_t modelId, uint16_t tagId, rtStream_t stream_);
RTS_API rtError_t rtProfilerTraceEx(uint64_t id, uint64_t modelId, uint16_t tagId, rtStream_t stream);

/**
* @ingroup profiling_base


Loading…
Cancel
Save