github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-26 17:53:08 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
07f5a4e054e183f4604734b174fda5a84c5cd652
ColossalAI/colossalai/fx/passes/algorithms
History
Boyuan Yao ac3a453a50 [fx] fix the discretize bug (#1506) 
* [fx] fix wrong variable name in solver rotor

* [fx] fix wrong variable name in solver rotor

* code modification

* [fx] fix the discretize bug
2022-08-26 17:15:52 +08:00
..
__init__.py
[fx] Add activation checkpoint solver rotor (#1496)
2022-08-26 10:34:21 +08:00
ckpt_solver_chen.py
[fx] add profiler for fx nodes. (#1480)
2022-08-24 16:22:44 +08:00
ckpt_solver_rotor.py
[fx] fix the discretize bug (#1506)
2022-08-26 17:15:52 +08:00
linearize.py
[fx] Add activation checkpoint solver rotor (#1496)
2022-08-26 10:34:21 +08:00
utils.py
[fx] fix wrong variable name in solver rotor (#1502)
2022-08-26 15:47:08 +08:00