ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI

History

Jiarui Fang 504ff1d101 [embeddings] use cache_ratio instead of cuda_row_num (#1611 )		2022-09-20 14:33:04 +08:00
..
cache_embedding	[embeddings] use cache_ratio instead of cuda_row_num (#1611 )	2022-09-20 14:33:04 +08:00
__init__.py	…
colo_module.py	…
embedding.py	…
linear.py	…
module_utils.py	…