saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-02 23:39:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add provider information for kernel in perf test. (#1082)

Browse source
This commit is contained in:
Zhang Lei 2019-05-22 21:35:11 -07:00 committed by jywu-msft
parent a6a4fef677
commit 2a07dc2fbf
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/core/framework/parallel_executor.cc
View file

@ -187,7 +187,7 @@ void ParallelExecutor::RunNodeAsyncInternal(size_t p_node_index,
session_state.Profiler().EndTimeAndRecordEvent(profiling::NODE_EVENT,
p_op_kernel->Node().Name() + "_kernel_time",
kernel_begin_time,
{{"op_name", p_op_kernel->KernelDef().OpName()}});
{{"op_name", p_op_kernel->KernelDef().OpName()}, {"provider", p_op_kernel->KernelDef().Provider()}});
sync_time_begin = session_state.Profiler().StartTime();
}

2
onnxruntime/core/framework/sequential_executor.cc
View file

@ -128,7 +128,7 @@ Status SequentialExecutor::Execute(const SessionState& session_state, const std:
session_state.Profiler().EndTimeAndRecordEvent(profiling::NODE_EVENT,
p_op_kernel->Node().Name() + "_kernel_time",
kernel_begin_time,
{{"op_name", p_op_kernel->KernelDef().OpName()}});
{{"op_name", p_op_kernel->KernelDef().OpName()}, {"provider", p_op_kernel->KernelDef().Provider()}});
sync_time_begin = session_state.Profiler().StartTime();
}