()
| 88 | |
| 89 | |
| 90 | def test_simple_auto_add_dfvar(): |
| 91 | root_fn = Identity["main"] |
| 92 | dfb = root_fn.body.blocks[0] |
| 93 | |
| 94 | rwt = DataflowBlockRewrite(dfb, root_fn) |
| 95 | rwt.add(root_fn.params[0], is_dfvar=True) |
| 96 | |
| 97 | assert isinstance(rwt.mutated_dfb().bindings[-1].var, DataflowVar) |
| 98 | |
| 99 | # immutatbility |
| 100 | assert_immutability(rwt, dfb, root_fn) |
| 101 | |
| 102 | |
| 103 | def test_simple_remove_unused(): |
nothing calls this directly
no test coverage detected
searching dependent graphs…