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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-13 03:38:19 +00:00

Files

YH 1a229045af Add interface for colo tesnor dp size (#3227 )

2023-03-27 09:42:21 +08:00

13 KiB

Raw Blame History

View Raw