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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-14 05:10:14 +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