提交 75ea577f 编写于 作者: X Xin Pan

allow profiler and timeline to work when dev_ctx is nullptr.

Sometimes dev_ctx is not available when RecordEvent.
上级 82e5738f
...@@ -127,6 +127,7 @@ double Event::CpuElapsedMs(const Event& e) const { ...@@ -127,6 +127,7 @@ double Event::CpuElapsedMs(const Event& e) const {
double Event::CudaElapsedMs(const Event& e) const { double Event::CudaElapsedMs(const Event& e) const {
#ifdef PADDLE_WITH_CUDA #ifdef PADDLE_WITH_CUDA
if (!has_cuda_) return 0.0;
PADDLE_ENFORCE(e.has_cuda() && has_cuda()); PADDLE_ENFORCE(e.has_cuda() && has_cuda());
PADDLE_ENFORCE(e.device() == device()); PADDLE_ENFORCE(e.device() == device());
PADDLE_ENFORCE(cudaEventSynchronize(event_)); PADDLE_ENFORCE(cudaEventSynchronize(event_));
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册