github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-03 09:23:58 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
c6930d8ddfc5e288375da8d0bb3f6a97d7f3f708
ColossalAI/colossalai/nn/layer/parallel_2p5d/_operation.py
Ziyue Jiang 4b01da24cd [TP] change the check assert in split batch 2d (#772)
2022-04-16 21:29:57 +08:00

37 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink