ColossalAI/colossalai/nn/parallel/layers/cache_embedding at 9a9ef65313f0fcdc4ab053348d5b3e02fa2e8c6a - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-04 09:52:50 +00:00

Files

History

Jiarui Fang 9a9ef65313 [FAW] cpu caching operations (#1520 )

2022-08-30 14:50:02 +08:00

..

__init__.py

[FAW] init an LFU implementation for FAW (#1488 )

2022-08-24 17:37:22 +08:00

base_embedding.py

[FAW] export FAW in _ops (#1438 )

2022-08-11 13:43:24 +08:00

cache_mgr.py

[FAW] cpu caching operations (#1520 )

2022-08-30 14:50:02 +08:00

copyer.py

[NFC] polish doc style for ColoTensor (#1457 )

2022-08-16 09:21:05 +08:00

freq_aware_embedding.py

[FAW] cpu caching operations (#1520 )

2022-08-30 14:50:02 +08:00

parallel_freq_aware_embedding.py

[FAW] cpu caching operations (#1520 )

2022-08-30 14:50:02 +08:00