!9776 fix heterogeneity SyncHostToDevice failed

From: @zhaosida1994 Reviewed-by: @chujinjin,@zhoufeng54 Signed-off-by: @chujinjin
4 years ago · 8bd2143500
parent eb8cdd8a96 89dccd8b9d
commit 8bd2143500
1 changed files with 1 additions and 1 deletions
--- a/mindspore/ccsrc/runtime/device/ascend/ascend_device_address.cc
+++ b/mindspore/ccsrc/runtime/device/ascend/ascend_device_address.cc
@ -575,7 +575,7 @@ bool AscendDeviceAddress::ConvertFormatAndSyncHostToDevice(const ShapeVector &sh
    host_shape.emplace_back(1);
  }
  std::vector<size_t> device_shape;
-  if (format_ == kOpFormat_FRAC_NZ || format_ == kOpFormat_NCDHW || kOpFormat_NDC1HWC0) {
+  if (format_ == kOpFormat_FRAC_NZ || format_ == kOpFormat_NCDHW || format_ == kOpFormat_NDC1HWC0) {
    device_shape = trans::TransShapeToDevice(host_shape, format_);
  } else {
    host_shape = trans::PaddingShapeTo4d(host_shape);