ColossalAI/tests/test_layers/test_cache_embedding.py at 521078ffc9f1a4bc8fc91e125ffd7dac683094cb - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-01 12:24:27 +00:00

Files

CsRic 5156d5b4f8 [embedding] add tablewise sharding for FAW (#1526 )

2022-09-01 17:55:41 +08:00

14 KiB

Raw Blame History

View Raw