未验证 提交 acd7309e 编写于 作者: Q Qiao Longfei 提交者: GitHub

Merge pull request #10079 from jacquesqiao/fix-reduce_op_handle_test

fix reduce_op_handle_test
...@@ -109,7 +109,7 @@ struct TestReduceOpHandle { ...@@ -109,7 +109,7 @@ struct TestReduceOpHandle {
// add input // add input
for (size_t j = 0; j < gpu_list_.size(); ++j) { for (size_t j = 0; j < gpu_list_.size(); ++j) {
if (!use_gpu_) { 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]); auto *in_var_handle = new VarHandle(1, j, "input", gpu_list_[j]);
in_var_handle->generated_op_ = nullptr; in_var_handle->generated_op_ = nullptr;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册