Commits · fa5f45f0a2a69b726b87e4d88d9a954eba804b5e · OpenDAS / FastMoE

29 May, 2021 1 commit
- fix bugs to run megatron with gshard gate · fa5f45f0
  Rick Ho authored May 29, 2021
  
  fa5f45f0
19 May, 2021 1 commit
- remove local scatter and gather in cuda · 52e57316
  Rick Ho authored May 19, 2021
  
  52e57316
18 May, 2021 1 commit
- use customized pos assignment kernel to support -1 · 980cf4b6
  Rick Ho authored May 18, 2021
  
  980cf4b6
27 Apr, 2021 2 commits
- gshard prune gate cuda code and test · 56cb8c15
  Rick Ho authored Apr 27, 2021
  
  56cb8c15
- limit by threshold cuda code · bb92d30e
  Rick Ho authored Apr 27, 2021
  
  bb92d30e
26 Apr, 2021 1 commit
- reconstruct fmoe cuda code · bb3c8966
  Rick Ho authored Apr 26, 2021
  
  bb3c8966