mirror of https://github.com/hpcaitech/ColossalAI
![]() * [fx] change memory.py to memory_utils.py. * [fx] add shard utils. * [fx] fix import. * [fx] check code style. * [fx] add comment. * [autoparallel] first move. * [fx] add time computations. |
||
---|---|---|
.. | ||
codegen | ||
passes | ||
profiler | ||
tracer | ||
__init__.py | ||
_compatibility.py | ||
_meta_registrations.py | ||
graph_module.py | ||
proxy.py |