Commits · 38c68b5b9adff9e30b0ac487b79d1b0b9f0c6fd0 · OpenDAS / ColossalAI

22 Sep, 2022 1 commit
- [embedding] rollback for better FAW performance (#1625) · 38c68b5b
  Jiarui Fang authored Sep 22, 2022
  
  38c68b5b
20 Sep, 2022 1 commit
- [embeddings] use cache_ratio instead of cuda_row_num (#1611) · 504ff1d1
  Jiarui Fang authored Sep 20, 2022
  
  504ff1d1
13 Sep, 2022 1 commit
- [embeddings] add already_split_along_rank flag for tablewise mode (#1584) · f3403ff9
  CsRic authored Sep 13, 2022
  
  f3403ff9
06 Sep, 2022 1 commit
- [embedding] polish parallel embedding tablewise (#1545) · 64169f3e
  Jiarui Fang authored Sep 06, 2022
  
  64169f3e
05 Sep, 2022 1 commit
- [embedding] freq_aware_embedding: add small functions for caller application (#1537) · 964123ae
  CsRic authored Sep 05, 2022
  
  964123ae
01 Sep, 2022 1 commit
- [embedding] add tablewise sharding for FAW (#1526) · 5156d5b4
  CsRic authored Sep 01, 2022
  
  5156d5b4
30 Aug, 2022 1 commit
- [FAW] cpu caching operations (#1520) · 9a9ef653
  Jiarui Fang authored Aug 30, 2022
  
  9a9ef653
29 Aug, 2022 2 commits
- [FAW] refactor reorder() for CachedParamMgr (#1514) · af5438ca
  Jiarui Fang authored Aug 29, 2022
  
  af5438ca
- [FAW] shrink freq_cnter size (#1509) · 1b8fee8e
  CsRic authored Aug 29, 2022
  
  1b8fee8e
26 Aug, 2022 1 commit
- [FAW] FAW embedding use LRU as eviction strategy intialized with dataset stats (#1494) · 0ed2f461
  CsRic authored Aug 26, 2022
  
  0ed2f461
25 Aug, 2022 1 commit
- [FAW] LFU cache for the FAW · b8d0e39e
  CsRic authored Aug 25, 2022
  
  b8d0e39e
24 Aug, 2022 1 commit
- [FAW] init an LFU implementation for FAW (#1488) · cde7b8a5
  Jiarui Fang authored Aug 24, 2022
  
  cde7b8a5
23 Aug, 2022 1 commit
- [FCE] update interface for frequency statistics in FreqCacheEmbedding (#1462) · 0aad53c6
  Geng Zhang authored Aug 23, 2022
  
  0aad53c6
12 Aug, 2022 1 commit
- [FAW] reorganize the inheritance struct of FreqCacheEmbedding (#1448) · 9f3eed66
  Geng Zhang authored Aug 12, 2022
  
  9f3eed66
11 Aug, 2022 1 commit
- [FAW] export FAW in _ops (#1438) · 30b4dd17
  Jiarui Fang authored Aug 11, 2022
  
  30b4dd17
10 Aug, 2022 3 commits
- [polish] add test_ops directory (#1431) · 89c434a0
  Jiarui Fang authored Aug 10, 2022
  
  89c434a0
- [FAW] move coloparam setting in test code. (#1429) · 10b3df65
  Jiarui Fang authored Aug 10, 2022
  
  10b3df65
- [FAW] parallel FreqAwareEmbedding (#1424) · cb98cf55
  Jiarui Fang authored Aug 10, 2022
  
  cb98cf55
09 Aug, 2022 2 commits
- Add FreqAwareEmbeddingBag (#1421) · d209aff6
  Jiarui Fang authored Aug 09, 2022
  
  d209aff6
- [FAW] add cache manager for the cached embedding (#1419) · 504419d2
  Jiarui Fang authored Aug 09, 2022
  
  504419d2