提交 d55a03d9 编写于 作者: Y Yu Yang

Scale loss on place

上级 932364a2
......@@ -146,6 +146,7 @@ struct ScaleLossGradOpHandle : public OpHandle {
if (platform::is_cpu_place(place_)) {
*tmp = coeff_;
} else {
VLOG(3) << "Scale loss on place" << place_;
auto stream =
static_cast<platform::CUDADeviceContext *>(this->dev_ctx_[place_])
->stream();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册