ColossalAI/colossalai/fx/codegen
duanjunwen 1b76564e16
[test] Fix/fix testcase (#5770)
* [fix] branch for fix testcase;

* [fix] fix test_analyzer & test_auto_parallel;

* [fix] remove local change about moe;

* [fix] rm local change moe;
2024-06-03 15:26:01 +08:00
..
__init__.py [fx] added activation checkpoint codegen support for torch < 1.12 (#1359) 2022-07-25 23:35:31 +08:00
activation_checkpoint_codegen.py [test] Fix/fix testcase (#5770) 2024-06-03 15:26:01 +08:00