From ad784cbe188de1bbf06454078cce6f6f73342872 Mon Sep 17 00:00:00 2001 From: luoyuan Date: Sat, 7 May 2022 11:15:38 +0800 Subject: [PATCH] [MS][LITE] fix copy fp64 tensordata from tensorinfo --- mindspore/lite/tools/optimizer/common/gllo_utils.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/lite/tools/optimizer/common/gllo_utils.cc b/mindspore/lite/tools/optimizer/common/gllo_utils.cc index 89c00aee015..6a8b12248ad 100644 --- a/mindspore/lite/tools/optimizer/common/gllo_utils.cc +++ b/mindspore/lite/tools/optimizer/common/gllo_utils.cc @@ -221,9 +221,9 @@ int CopyTensorDataFromTensorInfo(const tensor::TensorPtr &tensor_info, } auto *origin_data = reinterpret_cast(tensor_info->data_c()); for (size_t i = 0; i < data_count; ++i) { - if (origin_data[i] > static_cast(FLT_MAX) || origin_data[i] < static_cast(FLT_MIN)) { + if (origin_data[i] > static_cast(FLT_MAX) || origin_data[i] < static_cast(-FLT_MAX)) { MS_LOG(WARNING) << "float64 data " << origin_data[i] << " cannot fit into float32"; - tensor_data[i] = origin_data[i] > 0 ? FLT_MAX : FLT_MIN; + tensor_data[i] = origin_data[i] > 0 ? FLT_MAX : -FLT_MAX; } else { tensor_data[i] = static_cast(origin_data[i]); }