!1364 fix summary bug of gradient

Merge pull request !1364 from zyli2020/fix_summary_bug
pull/1364/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 479f26ea0d

@ -785,6 +785,9 @@ void SessionBasic::Summary(KernelGraph *graph) {
(void)std::copy(shape.begin(), shape.end(), std::back_inserter(temp_shape));
tensor::TensorPtr tensor = std::make_shared<tensor::Tensor>(type_id, temp_shape);
MS_EXCEPTION_IF_NULL(address);
if (!address->GetPtr()) {
continue;
}
if (!address->SyncDeviceToHost(trans::GetRuntimePaddingShape(node, index), LongToSize(tensor->data().nbytes()),
tensor->data_type(), tensor->data_c(true))) {
MS_LOG(ERROR) << "Failed to sync output from device to host.";

Loading…
Cancel
Save