ColossalAI

Author	SHA1	Message	Date
Jiarui Fang	c638bec028	[embedding] polish async copy (#1657 )	2022-09-27 14:37:03 +08:00
CsRic	0767f67a0f	[embedding] isolate cache_op from forward (#1645 ) Co-authored-by: ric <mkkt_bkkt@mail.ustc.edu.cn>	2022-09-26 11:18:59 +08:00
Jiarui Fang	e57df80325	[embeddings] cache option (#1635 )	2022-09-23 16:40:18 +08:00
Jiarui Fang	504ff1d101	[embeddings] use cache_ratio instead of cuda_row_num (#1611 )	2022-09-20 14:33:04 +08:00
Jiarui Fang	a19eb80998	[embedding] updates some default parameters	2022-09-15 15:45:17 +08:00
CsRic	5156d5b4f8	[embedding] add tablewise sharding for FAW (#1526 )	2022-09-01 17:55:41 +08:00
Jiarui Fang	4537d39df9	[doc] docstring for FreqAwareEmbeddingBag (#1525 )	2022-08-31 13:52:30 +08:00
Jiarui Fang	9a9ef65313	[FAW] cpu caching operations (#1520 )	2022-08-30 14:50:02 +08:00
Jiarui Fang	ba61109b6c	[FAW] remove code related to chunk (#1501 )	2022-08-26 14:23:30 +08:00
Jiarui Fang	cde7b8a5b8	[FAW] init an LFU implementation for FAW (#1488 )	2022-08-24 17:37:22 +08:00
Geng Zhang	0aad53c62b	[FCE] update interface for frequency statistics in FreqCacheEmbedding (#1462 )	2022-08-23 17:38:24 +08:00
Jiarui Fang	a1476ea882	[NFC] polish doc style for ColoTensor (#1457 )	2022-08-16 09:21:05 +08:00
Geng Zhang	9f3eed66eb	[FAW] reorganize the inheritance struct of FreqCacheEmbedding (#1448 )	2022-08-12 15:55:46 +08:00
Jiarui Fang	30b4dd17c0	[FAW] export FAW in _ops (#1438 )	2022-08-11 13:43:24 +08:00