From 570be3919543432682085335c03bdb46c0db6282 Mon Sep 17 00:00:00 2001 From: qiaolongfei Date: Fri, 20 Apr 2018 13:32:53 +0800 Subject: [PATCH] fix build --- paddle/fluid/framework/details/reduce_op_handle_test.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/framework/details/reduce_op_handle_test.cc b/paddle/fluid/framework/details/reduce_op_handle_test.cc index ed6a1355a3a..a57c7882ed7 100644 --- a/paddle/fluid/framework/details/reduce_op_handle_test.cc +++ b/paddle/fluid/framework/details/reduce_op_handle_test.cc @@ -109,7 +109,7 @@ struct TestReduceOpHandle { // add input for (size_t j = 0; j < gpu_list_.size(); ++j) { if (!use_gpu_) { - op_handle_->dev_ctxes_[gpu_list_[j]] = ctxs_[j].get(); + op_handle_->SetDeviceContext(gpu_list_[j], ctxs_[j].get()); } auto *in_var_handle = new VarHandle(1, j, "input", gpu_list_[j]); in_var_handle->generated_op_ = nullptr; -- GitLab