ColossalAI/colossalai/tensor/comm_spec.py at 1653063fce7062dfd652e73691fac44a24698cd5 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-25 01:03:35 +00:00

Files

YuliangLiu0306 47fb214b3b [hotfix] add shard dim to aviod backward communication error (#2954 )

2023-03-01 11:41:53 +08:00

22 KiB

Raw Blame History

View Raw