Lichen455
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix codex

This commit is contained in:
fary86 2020-06-05 10:36:08 +08:00
parent 05c1b36c51
commit 3bf1bb6355
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mindspore/ccsrc/pipeline/static_analysis/prim.cc
View File

@ -274,14 +274,15 @@ AnfNodePtr MixedPrecisionCastHelper(AnfNodePtr source_node, AbstractBasePtr node
} else if (node_type->isa<AbstractTuple>()) {
auto x = node_type->cast<AbstractTuplePtr>();
auto &items = x->elements();
std::size_t size = items.size();
std::vector<AnfNodePtr> nodes;
nodes.emplace_back(NewValueNode(prim::kPrimMakeTuple));
for (int i = 0; i < SizeToInt(size); i++) {
int idx = 0;
for (const auto &item : items) {
AnfNodePtr tuple_node =
func_graph->NewCNode({NewValueNode(prim::kPrimTupleGetItem), source_node, NewValueNode(i)});
AnfNodePtr node = MixedPrecisionCastHelper(tuple_node, items[i], target_type, func_graph);
func_graph->NewCNode({NewValueNode(prim::kPrimTupleGetItem), source_node, NewValueNode(idx)});
AnfNodePtr node = MixedPrecisionCastHelper(tuple_node, item, target_type, func_graph);
nodes.emplace_back(node);
++idx;
}
target_node = func_graph->NewCNode(nodes);
}

1
mindspore/ccsrc/utils/union_find_set.h
View File

@ -27,6 +27,7 @@ template <class T>
class UnionFindSet {
public:
UnionFindSet() : union_find_set_() {}
~UnionFindSet() = default;
void Add(const T &elem) {
if (union_find_set_.find(elem) != union_find_set_.end()) {
return;