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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-11 00:56:38 +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