Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
FastMoE
Commits
8f1f2ca54edf88bb4b0d1a4260967a8a8b3eef8e
Switch branch/tag
fastmoe
fmoe
functions.py
05 Feb, 2021
2 commits
pass pylint
· f2040d9f
Rick Ho
authored
Feb 05, 2021
f2040d9f
support multiple pytorch versions prviate apis
· bf2fd0c0
Rick Ho
authored
Feb 05, 2021
bf2fd0c0
04 Feb, 2021
3 commits
adapt with pytorch 1.8.0 (deprecated 1.6.0)
· 15f98a10
Rick Ho
authored
Feb 04, 2021
15f98a10
setup pylint and write docs for functions
· 585604fe
Rick Ho
authored
Feb 04, 2021
585604fe
fix no grad after all-gather bug
· 56c1bd63
Rick Ho
authored
Feb 04, 2021
56c1bd63
02 Feb, 2021
1 commit
Format using black and add model_parallel_rank
· fdbac1df
Sengxian
authored
Feb 02, 2021
fdbac1df
01 Feb, 2021
3 commits
split more tests
· 01d9b418
Rick Ho
authored
Feb 01, 2021
01d9b418
complete test for reconstruction
· 22e1eb45
Rick Ho
authored
Feb 01, 2021
22e1eb45
swap local scatter and gather kernel functions
· d2039fc7
Rick Ho
authored
Feb 01, 2021
d2039fc7
29 Jan, 2021
4 commits
fix another bug to make global moe run
· 14c0eab4
Rick Ho
authored
Jan 29, 2021
14c0eab4
fix more bugs to make the layers run in the model
· 0fea2991
Rick Ho
authored
Jan 29, 2021
0fea2991
fix python bugs
· 6900f1de
Rick Ho
authored
Jan 29, 2021
6900f1de
split fmoe functions
· 5e0af68d
Rick Ho
authored
Jan 29, 2021
5e0af68d
