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

github/ColossalAI

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