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(); }