Browse Source

!905 Bugfix: check pricision loss when cast from int64 to bool

From: @hugo1
Reviewed-by: @xchu42,@wqtshg
Signed-off-by: @ji_chen
tags/v1.2.0
mindspore-ci-bot Gitee 3 years ago
parent
commit
6d71f7ad1d
1 changed files with 4 additions and 1 deletions
  1. +4
    -1
      ge/graph/common/transop_util.cc

+ 4
- 1
ge/graph/common/transop_util.cc View File

@@ -23,7 +23,10 @@
namespace { namespace {
const int kInvalidTransopDataIndex = -1; const int kInvalidTransopDataIndex = -1;
const int kTransOpOutIndex = 0; const int kTransOpOutIndex = 0;
std::map<ge::DataType, ge::DataType> precision_loss_transfer_map = {{ge::DT_FLOAT, ge::DT_BOOL}};
std::map<ge::DataType, ge::DataType> precision_loss_transfer_map = {
{ge::DT_FLOAT, ge::DT_BOOL},
{ge::DT_INT64, ge::DT_BOOL}
};
} // namespace } // namespace


namespace ge { namespace ge {


Loading…
Cancel
Save