Commits · c77f676d31245283de408747b1696da67684ca24 · OpenDAS / FastMoE

29 May, 2021 1 commit
- fix bugs to run megatron with gshard gate · fa5f45f0
  Rick Ho authored May 29, 2021
  
  fa5f45f0
24 May, 2021 1 commit
- add random routing in gshard gate · e58e7b3b
  Rich Ho authored May 24, 2021
  
  e58e7b3b
21 May, 2021 1 commit
- update accoding to comments · 8d14dd29
  Rich Ho authored May 21, 2021
  
  8d14dd29
18 May, 2021 2 commits
- use customized pos assignment kernel to support -1 · 980cf4b6
  Rick Ho authored May 18, 2021
  
  980cf4b6
- fix bugs to run on multiple gpus · a468db2b
  Rick Ho authored May 18, 2021
  
  a468db2b
13 May, 2021 3 commits
- test switch gate · 38b334cc
  Rich Ho authored May 13, 2021
  
  38b334cc
- gshard gate test · ddfaaf49
  Rich Ho authored May 13, 2021
  
  ddfaaf49
- add test but cannot pass · 5a0ba835
  Rick Ho authored May 13, 2021
  
  5a0ba835
27 Apr, 2021 2 commits
- gshard prune gate cuda code and test · 56cb8c15
  Rick Ho authored Apr 27, 2021
  
  56cb8c15
- balancing cuda code · 1aced6d8
  Rick Ho authored Apr 27, 2021
  
  1aced6d8
26 Apr, 2021 3 commits
- fix zero gate and constant gate · 450549a4
  Rick Ho authored Apr 26, 2021
  
  450549a4
- add gshard and switch gate with loss · 4c90e6e8
  Rick Ho authored Apr 26, 2021
  
  4c90e6e8
- reconstruct gate structure · 82103edb
  Rick Ho authored Apr 26, 2021
  
  82103edb