github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'e8d8eda5e7'
${ noResults }
ColossalAI/colossalai/fx/__init__.py

4 lines
121 B
Raw Normal View History Unescape

[fx] support meta tracing for aten level computation graphs like functorch. (#1536) * [fx] support meta tracing for aten level computation graphs like functorch. * [fx] support meta tracing for aten level computation graphs like functorch. * [fx] remove redundant import. * [fx] add docstring.
2 years ago
from .tracer import ColoTracer, meta_trace
[fx] Add activation checkpoint solver rotor (#1496) * [fx] fix defining ckpt functions inside forward * [fx] Modify activation checkpoint codegen and add ColoGraphModule * [fx] some modification * some modifications * some modifications * some modifications * some modifications * some code modifications * [automatic_parallel] ckpt solver rotor * [fx] add ckpt_solver_rotor * [fx] modification * code refactor * code refactor
2 years ago
from .graph_module import ColoGraphModule
[fx/profiler] provide a table of summary. (#1634) * [fx/profiling] provide summary for MetaInfoProp. * [fx/profiler] provide a table of summary. * [fx] optimize table repr.
2 years ago
from .passes import MetaInfoProp