From 4452ff76b79b3a9acdcd15ba6e751117889db3fb Mon Sep 17 00:00:00 2001 From: Yu Yang Date: Fri, 13 Apr 2018 17:38:43 +0800 Subject: [PATCH] Fix CPU compile --- paddle/fluid/framework/details/op_handle_base.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/paddle/fluid/framework/details/op_handle_base.cc b/paddle/fluid/framework/details/op_handle_base.cc index 28f1e7b5088..534d77860f8 100644 --- a/paddle/fluid/framework/details/op_handle_base.cc +++ b/paddle/fluid/framework/details/op_handle_base.cc @@ -109,18 +109,18 @@ void OpHandleBase::RunAndRecordEvent(const std::function &callback) { void OpHandleBase::RunAndRecordEvent(platform::Place p, const std::function &callback) { +#ifdef PADDLE_WITH_CUDA if (platform::is_cpu_place(p) || events_.empty()) { callback(); } else { -#ifdef PADDLE_WITH_CUDA auto *ctx = dev_ctxes_.at(p); auto *cuda_ctx = static_cast(ctx); cuda_ctx->RecordEvent(events_.at(boost::get(p).device), callback); + } #else - PADDLE_THROW("Not implemented"); + callback(); #endif - } } } // namespace details -- GitLab