tests/test_fx/test_tracer/test_control_flow.py · 62f4e2eb0760ac8bfe28834b061dbc2bda93ade9 · OpenDAS / ColossalAI · GitLab

"tests/test_zero/test_sharded_optim_v2.py" did not exist on "bb2790cf0bcea11c07e16e077c51b1fd11b545b1"

Find file Blame History Permalink

[fx] supported data-dependent control flow in model tracing (#1185) · 6d86f1bc
Frank Lee authored Jun 29, 2022
```
* [fx] supported data-dependent control flow in model tracing

* polish code
```
6d86f1bc

test_control_flow.py 1.79 KB