mirror of https://github.com/hpcaitech/ColossalAI
0584654c79
* [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. |
||
---|---|---|
.. | ||
gpt_utils.py | ||
test_profiler_meta_info_prop.py |