Merge ascend_optimizer and ascend_parser. (#30776)

Merge ascend_optimizer and ascend_parser.
revert-31562-mean
dingsiyu 4 years ago committed by GitHub
parent 636fefd9f8
commit 4a26729540
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -130,7 +130,7 @@ class AscendIRParser(object):
name = e.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":
# ge_out_operator.append(self.var2geop["reduce_sum_0.tmp_0@GRAD"])
@ -233,6 +233,7 @@ class AscendOptimizer(Optimizer):
self.parser = AscendIRParser()
input_varlist = self._get_input_varlist(main_block.program)
startup_graph, main_graph = self.parser.parse_program(
startup_program, main_block.program, input_varlist, self.fetch_list)

Loading…
Cancel
Save