Merge ascend_optimizer and ascend_parser. (#30776)

Merge ascend_optimizer and ascend_parser.
4 years ago · 4a26729540
parent 636fefd9f8
commit 4a26729540
2 changed files with 1602 additions and 254 deletions
--- a/python/paddle/distributed/fleet/meta_optimizers/ascend/ascend_optimizer.py
+++ b/python/paddle/distributed/fleet/meta_optimizers/ascend/ascend_optimizer.py
@ -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)

--- a/python/paddle/distributed/fleet/meta_optimizers/ascend/ascend_parser.py
+++ b/python/paddle/distributed/fleet/meta_optimizers/ascend/ascend_parser.py