Merge pull request #10079 from jacquesqiao/fix-reduce_op_handle_test

fix reduce_op_handle_test
wangkuiyi-patch-2
Qiao Longfei 7 years ago committed by GitHub
commit acd7309ea1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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

Loading…
Cancel
Save