sfqcyy
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!27 add log for kernel runtime in order to trace performance 

Merge pull request !27 from shibeiji/master
This commit is contained in:
mindspore-ci-bot 2020-03-31 11:13:27 +08:00 committed by Gitee
parent 140a15924c 468e257a14
commit 84a61bd015
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mindspore/ccsrc/device/kernel_runtime.cc
View File

@ -50,12 +50,19 @@ bool KernelRuntime::Run(session::KernelGraph *graph) {
bool ret = false; bool ret = false;
auto context_ptr = MsContext::GetInstance(); auto context_ptr = MsContext::GetInstance();
MS_EXCEPTION_IF_NULL(context_ptr); MS_EXCEPTION_IF_NULL(context_ptr);
struct timeval start_time, end_time;
(void)gettimeofday(&start_time, nullptr);
bool is_task_sink = context_ptr->enable_task_sink(); bool is_task_sink = context_ptr->enable_task_sink();
if (is_task_sink) { if (is_task_sink) {
ret = RunTask(graph); ret = RunTask(graph);
} else { } else {
ret = LaunchKernel(graph); ret = LaunchKernel(graph);
} }
(void)gettimeofday(&end_time, nullptr);
const uint64_t kUSecondInSecond = 1000000;
uint64_t cost = kUSecondInSecond * static_cast<uint64_t>(end_time.tv_sec - start_time.tv_sec);
cost += static_cast<uint64_t>(end_time.tv_usec - start_time.tv_usec);
MS_LOG(INFO) << "Call MS Run Success in " << cost << " us";
return ret; return ret;
} }