add fastmoe.megatron to install list

9a4434f5 · Rick Ho · 2bceb76b · 9a4434f5
Commit 9a4434f5 authored Mar 22, 2021 by Rick Ho
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

setup.py setup.py +1 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -20,7 +20,7 @@ if __name__ == '__main__':
        author_email='hja20@mails.tsinghua.edu.cn',
        license='Apache-2',
        url='https://github.com/laekov/fastmoe',
-        packages=['fmoe'],
+        packages=['fmoe', 'fmoe.megatron'],
        ext_modules=[
            CUDAExtension(
                name='fmoe_cuda',