提交 4452ff76 编写于 作者: Y Yu Yang

Fix CPU compile

上级 482314e3
...@@ -109,18 +109,18 @@ void OpHandleBase::RunAndRecordEvent(const std::function<void()> &callback) { ...@@ -109,18 +109,18 @@ void OpHandleBase::RunAndRecordEvent(const std::function<void()> &callback) {
void OpHandleBase::RunAndRecordEvent(platform::Place p, void OpHandleBase::RunAndRecordEvent(platform::Place p,
const std::function<void()> &callback) { const std::function<void()> &callback) {
#ifdef PADDLE_WITH_CUDA
if (platform::is_cpu_place(p) || events_.empty()) { if (platform::is_cpu_place(p) || events_.empty()) {
callback(); callback();
} else { } else {
#ifdef PADDLE_WITH_CUDA
auto *ctx = dev_ctxes_.at(p); auto *ctx = dev_ctxes_.at(p);
auto *cuda_ctx = static_cast<platform::CUDADeviceContext *>(ctx); auto *cuda_ctx = static_cast<platform::CUDADeviceContext *>(ctx);
cuda_ctx->RecordEvent(events_.at(boost::get<platform::CUDAPlace>(p).device), cuda_ctx->RecordEvent(events_.at(boost::get<platform::CUDAPlace>(p).device),
callback); callback);
}
#else #else
PADDLE_THROW("Not implemented"); callback();
#endif #endif
}
} }
} // namespace details } // namespace details
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册