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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-03 01:19:15 +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