ColossalAI/colossalai/nn/_ops/batch_norm.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 13:48:15 +00:00

Files

Jiarui Fang 8daf1b4db1 [Gemini] patch for supporting orch.add_ function for ColoTensor (#2003 )

2022-11-25 20:06:35 +08:00

1.1 KiB

Raw Blame History

View Raw