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

Set dev

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