17 Commits (b29e1f07224298aea35aab7ee83284beac28e0d8)

Author	SHA1	Message	Date
Frank Lee	80eba05b0a	[test] refactor tests with spawn (#3452) ... * [test] added spawn decorator * polish code * polish code * polish code * polish code * polish code * polish code	2023-04-06 14:51:35 +08:00
Jiarui Fang	21962e1593	[embedding] rename FreqAwareEmbedding -> CachedEmbedding (#1699)	2022-10-13 22:22:27 +08:00
Jiarui Fang	38c68b5b9a	[embedding] rollback for better FAW performance (#1625)	2022-09-22 11:16:25 +08:00
Jiarui Fang	504ff1d101	[embeddings] use cache_ratio instead of cuda_row_num (#1611)	2022-09-20 14:33:04 +08:00
CsRic	f3403ff98e	[embeddings] add already_split_along_rank flag for tablewise mode (#1584)	2022-09-13 10:50:34 +08:00
Jiarui Fang	64169f3e8f	[embedding] polish parallel embedding tablewise (#1545)	2022-09-06 10:41:20 +08:00
CsRic	964123ae0f	[embedding] freq_aware_embedding: add small functions for caller application (#1537)	2022-09-05 15:12:53 +08:00
CsRic	5156d5b4f8	[embedding] add tablewise sharding for FAW (#1526)	2022-09-01 17:55:41 +08:00
Jiarui Fang	9a9ef65313	[FAW] cpu caching operations (#1520)	2022-08-30 14:50:02 +08:00
Jiarui Fang	af5438caa2	[FAW] refactor reorder() for CachedParamMgr (#1514)	2022-08-29 14:22:07 +08:00
CsRic	1b8fee8e9c	[FAW] shrink freq_cnter size (#1509)	2022-08-29 11:44:55 +08:00
CsRic	0ed2f46131	[FAW] FAW embedding use LRU as eviction strategy intialized with dataset stats (#1494)	2022-08-26 11:24:12 +08:00
CsRic	b8d0e39eaf	[FAW] LFU cache for the FAW	2022-08-25 13:08:46 +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
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