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

Scale loss on place

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