p85620714
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!6740 fix the bug of download timeline in GPU show time is abnormal 

Merge pull request !6740 from gzhcv/GpuTimeline
This commit is contained in:
mindspore-ci-bot 2020-09-23 09:42:55 +08:00 committed by Gitee
parent 60e86d80d8 f0f4ee985b
commit 7f35e2e2a1
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/profiler/device/gpu/gpu_profiling.cc
View File

@ -406,8 +406,10 @@ void GPUProfiler::OpDataProducerBegin(const std::string op_name, void *stream) {
CHECK_CUDA_RET_WITH_ERROR(cudaEventRecord(op_event_start_, (CUstream)stream_),
"cudaEventRecord op event start failed");
op_host_time_start_ = GetHostTimeStamp();
op_cupti_time_start_ = GetCUPTITimeStamp();
} else {
op_host_time_start_ = GetHostTimeStamp();
op_cupti_time_start_ = GetCUPTITimeStamp();
}
SetRunTimeData(op_name, stream);
}
@ -431,7 +433,7 @@ void GPUProfiler::OpDataProducerEnd() {
}
MS_LOG(DEBUG) << "Host Time Elapsed(us)," << op_name_ << "," << op_time_elapsed;
SetRunTimeData(op_name_, op_time_elapsed);
SetRunTimeData(op_name_, op_host_time_start_, op_time_elapsed);
SetRunTimeData(op_name_, op_cupti_time_start_, op_time_elapsed);
}
void GPUProfiler::StopCUPTI() {

1
mindspore/ccsrc/profiler/device/gpu/gpu_profiling.h
View File

@ -170,6 +170,7 @@ class GPUProfiler {
cudaEvent_t op_event_stop_;
uint64_t op_host_time_start_;
uint64_t op_host_time_stop_;
uint64_t op_cupti_time_start_;
std::string profile_data_path_;
};
} // namespace gpu