ColossalAI/colossalai/device at 8b045b3c1f4e6b8bfae062f0318bd1481c881a10 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-19 09:04:50 +00:00

Files

History

YuliangLiu0306 677e1e20d4 [device] update flatten device mesh usage (#2079 )

2022-12-05 16:16:07 +08:00

..

__init__.py

[autoparallel] Add alpha beta (#1973 )

2022-11-17 16:01:14 +08:00

calc_pipeline_strategy.py

[autoparallel] Add alpha beta (#1973 )

2022-11-17 16:01:14 +08:00

device_mesh.py

[device] update flatten device mesh usage (#2079 )

2022-12-05 16:16:07 +08:00

profile_alpha_beta.py

[autoparallel] Add alpha beta (#1973 )

2022-11-17 16:01:14 +08:00