mirror of https://github.com/hpcaitech/ColossalAI
5d0fdb9cb4
* [fx] fix offload codegen test * [fx] modify typing |
||
|---|---|---|
| .. | ||
| test_activation_checkpoint_codegen.py | ||
| test_nested_activation_checkpoint_codegen.py | ||
| test_offload_codegen.py | ||