ColossalAI/tests/test_tensor/test_context.py at 4603538dddc7957bc3ebc29caa066471da2417ba - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-24 22:35:06 +00:00

Files

Jiarui Fang 52c6ad26e0 [ColoTensor] reconfig ColoInitContext, decouple default_pg and default_dist_spec. (#1953 )

2022-11-15 16:24:16 +08:00

2.5 KiB

Raw Blame History

View Raw