| 25 | |
| 26 | |
| 27 | class ReplaceVar(ExprMutator): |
| 28 | def __init__(self, old_var, new_var): |
| 29 | super().__init__() |
| 30 | self.old_var = old_var |
| 31 | self.new_var = new_var |
| 32 | |
| 33 | def visit_var_(self, op): |
| 34 | if op.same_as(self.old_var): |
| 35 | return self.new_var |
| 36 | return op |
| 37 | |
| 38 | |
| 39 | def test_expr_constructor(): |
no outgoing calls
searching dependent graphs…