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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-26 09:42:27 +00:00

Files

CsRic f3403ff98e [embeddings] add already_split_along_rank flag for tablewise mode (#1584 )

2022-09-13 10:50:34 +08:00

14 KiB

Raw Blame History

View Raw