mirror of https://github.com/hpcaitech/ColossalAI
f3687e4ee2
* [fx] add nested activation_checkpoint codegen * undo algorithms commits * solver * undo some commits * [fx] torch11 add nested activation checkpoint codegen * remove some imports * [fx] add some comments in activation codegen * [fx] codegen instance error fix |
||
---|---|---|
.. | ||
test_activation_checkpoint_codegen.py | ||
test_nested_activation_checkpoint_codegen.py |