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

fix rewrite controlflow bug

This commit is contained in:
yuzhenhua 2023-03-07 14:44:25 +08:00
parent 94ad76bc14
commit 29cfe21bd0
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/python/mindspore/rewrite/node.py
View File

@ -1126,7 +1126,7 @@ class Node:
elt.id = scoped_value.value
elif isinstance(elt, ast.Attribute) and isinstance(elt.value, ast.Name):
elt.value.id = scoped_value.scope
elt.value = scoped_value.value
elt.attr = scoped_value.value
else:
raise RuntimeError("Only support constant or symbol in tuple now")
else:

10
mindspore/python/mindspore/rewrite/parsers/for_parser.py
View File

@ -63,7 +63,7 @@ class ForParser(Parser):
logger.warning(
f"For MindSpore Rewrtie, illegal iteration condition for For node, it must start with{EVAL_WHITE_LIST}")
return
if iter_code.startswith("self"):
if "self" in iter_code:
iter_code = iter_code.replace("self", "stree.get_origin_network()")
try:
iter_obj = eval(iter_code)
@ -89,13 +89,13 @@ class ForParser(Parser):
stree.get_ast_root().body.remove(node)
return
if isinstance(iter_obj, range):
logger.warning("For MindSpore Rewrtie, range not support.")
logger.warning("For MindSpore Rewrite, range not support.")
elif isinstance(iter_obj, zip):
logger.warning("For MindSpore Rewrtie, zip not support.")
logger.warning("For MindSpore Rewrite, zip not support.")
elif isinstance(iter_obj, enumerate):
logger.warning("For MindSpore Rewrtie, enumerate not support.")
logger.warning("For MindSpore Rewrite, enumerate not support.")
else:
logger.warning("For MindSpore Rewrtie, not supported type: ", type(iter_obj))
logger.warning(f"For MindSpore Rewrite, not supported type: {type(iter_obj).__name__}")
stree.try_append_python_node(node, node)
return