ColossalAI/colossalai/utils/model
ver217 ad536e308e
[tensor] refactor colo-tensor (#992)
* refactor colo-tensor and update linear op

* polish code

* polish code

* update ops and unit tests

* update unit tests

* polish code

* rename dist_spec module

* polish code

* polish code

* remove unneeded import

* fix pipelinable
2022-05-19 12:44:59 +08:00
..
__init__.py fix import error (#880) 2022-04-26 19:28:40 +08:00
colo_init_context.py [tensor] refactor colo-tensor (#992) 2022-05-19 12:44:59 +08:00
pipelinable.py [tensor] refactor colo-tensor (#992) 2022-05-19 12:44:59 +08:00
utils.py [pipelinable]use pipelinable context to initialize non-pipeline model (#816) 2022-04-24 13:03:12 +08:00