未验证 提交 4a267295 编写于 作者: D dingsiyu 提交者: GitHub

Merge ascend_optimizer and ascend_parser. (#30776)

Merge ascend_optimizer and ascend_parser.
上级 636fefd9
...@@ -130,7 +130,7 @@ class AscendIRParser(object): ...@@ -130,7 +130,7 @@ class AscendIRParser(object):
name = e.name name = e.name
ge_out_operator.append(self.var2geop[name]) ge_out_operator.append(self.var2geop[name])
# (Debug) If you want to print back prop vars, append/assign the varname in ge_out_operator here, such as: # (Debug) If you want to print back prop vars, append/assign the varname in ge_out_operator here, such as:
# if graph_name == "main": # if graph_name == "main":
# ge_out_operator.append(self.var2geop["reduce_sum_0.tmp_0@GRAD"]) # ge_out_operator.append(self.var2geop["reduce_sum_0.tmp_0@GRAD"])
...@@ -233,6 +233,7 @@ class AscendOptimizer(Optimizer): ...@@ -233,6 +233,7 @@ class AscendOptimizer(Optimizer):
self.parser = AscendIRParser() self.parser = AscendIRParser()
input_varlist = self._get_input_varlist(main_block.program) input_varlist = self._get_input_varlist(main_block.program)
startup_graph, main_graph = self.parser.parse_program( startup_graph, main_graph = self.parser.parse_program(
startup_program, main_block.program, input_varlist, self.fetch_list) startup_program, main_block.program, input_varlist, self.fetch_list)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册