p83571920
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Modify Double Comparison

This commit is contained in:
Yang Jiao 2021-07-29 11:17:37 +08:00
parent 32e4d2971b
commit 042c95d3d5
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/ccsrc/backend/optimizer/graph_kernel/arithmetic_simplify.cc
View File

@ -176,7 +176,6 @@ graphkernel::NType PatternNodeType(const std::string &n) {
return graphkernel::NType::Primitive;
}
}
bool IsEqual(double a, double b) { return (abs(a - b) < 0.0000001); }
std::string CleanStr(const std::string &s) {
std::string res = "";
@ -213,13 +212,13 @@ bool CheckCurNode(const graphkernel::NodePtr &tmp_node, const std::string &tmp_p
auto pattern_value_str =
std::static_pointer_cast<graphkernel::ConstTensorNode>((*const_to_ref)[tmp_pattern_op])->ToString();
double pattern_value = std::stod(CleanStr(pattern_value_str));
if (!IsEqual(pattern_value, node_value)) return false;
if (pattern_value != node_value) return false;
} else {
(*const_to_ref)[tmp_pattern_op] = tmp_node;
}
} else {
double pattern_value = std::stod(tmp_pattern_op);
if (!IsEqual(pattern_value, node_value)) {
if (pattern_value != node_value) {
return false;
}
}