Scale loss on place

helinwang-patch-1
Yu Yang 7 years ago
parent 932364a275
commit d55a03d916

@ -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();

Loading…
Cancel
Save