ColossalAI/tests/test_tensor/test_zero_optim.py at 6b2f2ab9bb410b7edd722dbfe6ee425de1437c31 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-12 19:06:17 +00:00

Files

Jiarui Fang 4b9bba8116 [ColoTensor] rename APIs and add output_replicate to ComputeSpec (#1168 )

2022-06-24 13:08:54 +08:00

5.5 KiB

Raw Blame History

View Raw