Fix CPU compile

wangkuiyi-patch-2
Yu Yang 7 years ago
parent 482314e3b1
commit 4452ff76b7

@ -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

Loading…
Cancel
Save