Commits · 969ef607227740bd034a678a8855b90bf7ba15aa · OpenDAS / FastMoE

12 Jan, 2021 1 commit
- fix argument count bug in moe_test.py · 46e6342e
  Rick Ho authored Jan 12, 2021
  
  46e6342e
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 1 commit
- global scatter gather kernels and pytorch C function · ec322e4b
  Rick Ho authored Jan 10, 2021
  
  ec322e4b
09 Jan, 2021 3 commits
- stream manager across multiple devices · 60b93e39
  Rick Ho authored Jan 09, 2021
  
  60b93e39
- modify performance test · c91dfad8
  Rick Ho authored Jan 09, 2021
  
  c91dfad8
- update stream manager · 143e21cc
  Rick Ho authored Jan 09, 2021
  
  143e21cc
02 Jan, 2021 2 commits
- update · 20cc924b
  Jiezhong Qiu authored Jan 03, 2021
  
  20cc924b
- set device according to input · 046455a8
  Jiezhong Qiu authored Jan 03, 2021
  
  046455a8
01 Jan, 2021 1 commit
- moe test script · 2ed4351d
  Rick Ho authored Jan 01, 2021
  
  2ed4351d
31 Dec, 2020 2 commits
- general support for profiling · 706f0cfd
  Rick Ho authored Jan 01, 2021
  
  706f0cfd
- nccl runs with 2 gpus · 4cb75d42
  Rick Ho authored Dec 31, 2020
  
  4cb75d42
30 Dec, 2020 5 commits
- test support for fused fw and bw · 254ad118
  Rick Ho authored Dec 30, 2020
  
  254ad118
- better timing · 191c1e46
  Rick Ho authored Dec 30, 2020
  
  191c1e46
- fix timing bug · 0d9c540f
  Rick Ho authored Dec 30, 2020
  
  0d9c540f
- tide up C code · 8cff6ad7
  Rick Ho authored Dec 30, 2020
  
  8cff6ad7
- performance test in python · cb6aadaa
  Rick Ho authored Dec 30, 2020
  
  cb6aadaa