From 2a07dc2fbfbb9b83272256020f922779c7f783e2 Mon Sep 17 00:00:00 2001 From: Zhang Lei Date: Wed, 22 May 2019 21:35:11 -0700 Subject: [PATCH] Add provider information for kernel in perf test. (#1082) --- onnxruntime/core/framework/parallel_executor.cc | 2 +- onnxruntime/core/framework/sequential_executor.cc | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/onnxruntime/core/framework/parallel_executor.cc b/onnxruntime/core/framework/parallel_executor.cc index 053cf653d5..c754af708c 100644 --- a/onnxruntime/core/framework/parallel_executor.cc +++ b/onnxruntime/core/framework/parallel_executor.cc @@ -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(); } diff --git a/onnxruntime/core/framework/sequential_executor.cc b/onnxruntime/core/framework/sequential_executor.cc index 94100bc020..258a288069 100644 --- a/onnxruntime/core/framework/sequential_executor.cc +++ b/onnxruntime/core/framework/sequential_executor.cc @@ -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(); }