提交 99f85a9f 编写于 作者: Y Yu Yang

Set dev

上级 d26f093f
...@@ -131,6 +131,7 @@ struct ScaleLossGradOpHandle : public OpHandle { ...@@ -131,6 +131,7 @@ struct ScaleLossGradOpHandle : public OpHandle {
: coeff_(static_cast<float>(1.0 / num_dev)), : coeff_(static_cast<float>(1.0 / num_dev)),
scope_(scope), scope_(scope),
place_(place) { place_(place) {
cudaSetDevice(boost::get<platform::CUDAPlace>(place_).device);
PADDLE_ENFORCE(cudaEventCreateWithFlags(&ev_, cudaEventDisableTiming)); PADDLE_ENFORCE(cudaEventCreateWithFlags(&ev_, cudaEventDisableTiming));
VLOG(3) << "Create " << ev_; VLOG(3) << "Create " << ev_;
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册