This commit is contained in:
liubuyu 2020-04-18 09:49:12 +08:00
parent fc07cd908e
commit 852e61d46c
1 changed files with 1 additions and 1 deletions

View File

@ -117,7 +117,7 @@ bool AscendDeviceAddress::SyncDeviceToHost(const std::vector<int> &shape, size_t
} else {
auto iter = kNeedTransFormatSet.find(format_);
if (iter != kNeedTransFormatSet.end()) {
sync_ok = ConvertFormatAndSyncHostToDevice(shape, size, type, host_ptr);
sync_ok = SyncDeviceToHostAndConvertFormat(shape, size, type, host_ptr);
}
}
if (!sync_ok) {