[Submodule] Add FastMoE as third-party library (#298)

THUDM · Nov 5, 2021 · d94fe57 · d94fe57
1 parent 9a2aceb
commit d94fe57
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/.gitmodules b/.gitmodules
@@ -6,3 +6,7 @@
 	path = third_party/actnn
 	url = https://github.com/ucbrise/actnn
 	branch = main
+[submodule "third_party/fastmoe"]
+	path = third_party/fastmoe
+	url = https://github.com/laekov/fastmoe
+	branch = master
diff --git a/third_party/README.md b/third_party/README.md
@@ -4,6 +4,8 @@
 
 [ActNN](https://github.com/ucbrise/actnn) can reduce the training memory footprint by compressing the saved activations.
 
+[FastMoE](https://github.com/laekov/fastmoe) can be used for GNN models wtih the Mixture of Experts (MoE).
+
 ## Installation
 
 For dgNN,
@@ -17,3 +19,9 @@ For ActNN,
 cd actnn/actnn
 pip install -v -e .
 ```
+
+For FastMoE,
+```bash
+cd fastmoe
+python setup.py install
+```
diff --git a/third_party/fastmoe b/third_party/fastmoe