AST mutator printer using StmtExprMutator.
| 501 | |
| 502 | |
| 503 | class StmtExprMutatorPrinter(StmtExprMutator): |
| 504 | """AST mutator printer using StmtExprMutator.""" |
| 505 | |
| 506 | def __init__(self) -> None: |
| 507 | super().__init__() |
| 508 | self.log = ASTLog() |
| 509 | |
| 510 | def visit_bind_(self, op): |
| 511 | result = super().visit_bind_(op) |
| 512 | self.log.add("Bind") |
| 513 | return result |
| 514 | |
| 515 | def visit_attr_(self, op): |
| 516 | result = super().visit_attr_(op) |
| 517 | self.log.add("AttrStmt") |
| 518 | return result |
| 519 | |
| 520 | def visit_assert_(self, op): |
| 521 | result = super().visit_assert_(op) |
| 522 | self.log.add("AssertStmt") |
| 523 | return result |
| 524 | |
| 525 | def visit_for_(self, op): |
| 526 | result = super().visit_for_(op) |
| 527 | self.log.add("For") |
| 528 | return result |
| 529 | |
| 530 | def visit_while_(self, op): |
| 531 | result = super().visit_while_(op) |
| 532 | self.log.add("While") |
| 533 | return result |
| 534 | |
| 535 | def visit_buffer_store_(self, op): |
| 536 | result = super().visit_buffer_store_(op) |
| 537 | self.log.add("BufferStore") |
| 538 | return result |
| 539 | |
| 540 | def visit_seqstmt_(self, op): |
| 541 | result = super().visit_seqstmt_(op) |
| 542 | self.log.add("SeqStmt") |
| 543 | return result |
| 544 | |
| 545 | def visit_evaluate_(self, op): |
| 546 | result = super().visit_evaluate_(op) |
| 547 | self.log.add("Evaluate") |
| 548 | return result |
| 549 | |
| 550 | def visit_block_(self, op): |
| 551 | result = super().visit_block_(op) |
| 552 | self.log.add("Block") |
| 553 | return result |
| 554 | |
| 555 | def visit_block_realize_(self, op): |
| 556 | result = super().visit_block_realize_(op) |
| 557 | self.log.add("BlockRealize") |
| 558 | return result |
| 559 | |
| 560 | # ExprMutator methods |
no outgoing calls
searching dependent graphs…