Commits · 969ef607227740bd034a678a8855b90bf7ba15aa · OpenDAS / FastMoE

13 Jan, 2021 1 commit
- import from pytorch dir · 6b38250f
  Jiezhong Qiu authored Jan 13, 2021
  
  6b38250f
12 Jan, 2021 2 commits
- fix argument count bug in moe_test.py · 46e6342e
  Rick Ho authored Jan 12, 2021
  
  46e6342e
- add nccl flag in setup.py · 3501fa41
  Rick Ho authored Jan 12, 2021
  
  3501fa41
11 Jan, 2021 5 commits
- multi-gpu performance test · b9c28810
  Rick Ho authored Jan 11, 2021
  
  b9c28810
- distributed test weight · 76327544
  Rick Ho authored Jan 11, 2021
  
  76327544
- multi-gpu forward pass test · 864a4522
  Rick Ho authored Jan 11, 2021
  
  864a4522
- make moe run with cuda · 069cf01a
  Rick Ho authored Jan 11, 2021
  
  069cf01a
- split function file · a4f7f1da
  Rick Ho authored Jan 11, 2021
  
  a4f7f1da
10 Jan, 2021 3 commits
- global scatter gather kernels and pytorch C function · ec322e4b
  Rick Ho authored Jan 10, 2021
  
  ec322e4b
- use expert_count array instead of expert_n · 7a2ad4a1
  Rick Ho authored Jan 10, 2021
  
  7a2ad4a1
- remove weight input to c expert count function · 861b75c1
  Rick Ho authored Jan 10, 2021
  
  861b75c1
09 Jan, 2021 10 commits
- stream manager across multiple devices · 60b93e39
  Rick Ho authored Jan 09, 2021
  
  60b93e39
- make backward pass test · 27d8beaa
  Rick Ho authored Jan 09, 2021
  
  27d8beaa
- moe backward (cannot pass test) · 92f1774a
  Rick Ho authored Jan 09, 2021
  
  92f1774a
- modify performance test · c91dfad8
  Rick Ho authored Jan 09, 2021
  
  c91dfad8
- split operators and make forward run · 49732231
  Rick Ho authored Jan 09, 2021
  
  49732231
- update stream manager · 143e21cc
  Rick Ho authored Jan 09, 2021
  
  143e21cc
- stream manager fixed · 2565f2fa
  Rick Ho authored Jan 09, 2021
  
  2565f2fa
- fix bugs to pass test · b8a212ef
  Rick Ho authored Jan 09, 2021
  
  b8a212ef
- degrade to single fc fwd · 284f1424
  Rick Ho authored Jan 09, 2021
  
  284f1424
- fix last indent bug · cbd86de8
  Rick Ho authored Jan 09, 2021
  
  cbd86de8
04 Jan, 2021 1 commit
- stream manager object instead of pointer · ef83c893
  Jiezhong Qiu authored Jan 04, 2021
  
  ef83c893
03 Jan, 2021 6 commits
- fix memory leak · e52c0380
  Jiezhong Qiu authored Jan 03, 2021
  
  e52c0380
- cudafree malloced memory · 1bde1726
  Jiezhong Qiu authored Jan 03, 2021
  
  1bde1726
- update · 3be6ac10
  Jiezhong Qiu authored Jan 03, 2021
  
  3be6ac10
- update · 6212aa28
  Jiezhong Qiu authored Jan 03, 2021
  
  6212aa28
- make cudaStreamManager thread local · 27af1828
  Jiezhong Qiu authored Jan 03, 2021
  
  27af1828
- test data parallel (FAILED!) · f5cc759c
  Jiezhong Qiu authored Jan 03, 2021
  
  f5cc759c
02 Jan, 2021 5 commits
- CUDA error at moe_cuda_kernel.cu:86 code=13(CUBLAS_STATUS_EXECUTION_FAILED) · 85306aa7
  Jiezhong Qiu authored Jan 03, 2021
  
  85306aa7
- update · 20cc924b
  Jiezhong Qiu authored Jan 03, 2021
  
  20cc924b
- set device according to input · 046455a8
  Jiezhong Qiu authored Jan 03, 2021
  
  046455a8
- incoporate customized cuda moe input xfmr-xl · 2338a26e
  Jiezhong Qiu authored Jan 03, 2021
  
  2338a26e
- add moe debug flag · d036918c
  Jiezhong Qiu authored Jan 02, 2021
  
  d036918c
01 Jan, 2021 1 commit
- moe test script · 2ed4351d
  Rick Ho authored Jan 01, 2021
  
  2ed4351d
31 Dec, 2020 5 commits
- general support for profiling · 706f0cfd
  Rick Ho authored Jan 01, 2021
  
  706f0cfd
- make it run · 881b10c2
  Rick Ho authored Jan 01, 2021
  
  881b10c2
- fix ptr correctness · 691e92e1
  Rick Ho authored Jan 01, 2021
  
  691e92e1
- fix output bug · 7e949a62
  Rick Ho authored Dec 31, 2020
  
  7e949a62
- nccl runs with 2 gpus · 4cb75d42
  Rick Ho authored Dec 31, 2020
  
  4cb75d42
30 Dec, 2020 1 commit
- manual batched scatter and gather kernels · d690c7b2
  Rick Ho authored Dec 30, 2020
  
  d690c7b2