()
| 76 | |
| 77 | |
| 78 | def test_simple_auto_add_var(): |
| 79 | root_fn = Identity["main"] |
| 80 | dfb = root_fn.body.blocks[0] |
| 81 | |
| 82 | rwt = DataflowBlockRewrite(dfb, root_fn) |
| 83 | rwt.add(root_fn.params[0], is_dfvar=False) |
| 84 | |
| 85 | assert isinstance(rwt.mutated_dfb().bindings[-1].var, Var) |
| 86 | |
| 87 | assert_immutability(rwt, dfb, root_fn) |
| 88 | |
| 89 | |
| 90 | def test_simple_auto_add_dfvar(): |
nothing calls this directly
no test coverage detected
searching dependent graphs…