Skip to content

GitLab

You need to sign in or sign up before continuing.
Commit 7f15d11d authored by Rick Ho's avatar Rick Ho
Browse files

update version number

parent d205aaeb
Hide whitespace changes
Inline Side-by-side
Showing with 11 additions and 2 deletions
setup.py
View file @ 7f15d11d
...@@ -6,6 +6,15 @@ import os ...@@ -6,6 +6,15 @@ import os
cxx_flags = [] cxx_flags = []
ext_libs = [] ext_libs = []
authors = [
'Jiaao He',
'Jiezhong Qiu',
'Aohan Zeng',
'Tiago Antunes',
'Jinjun Peng',
'Qin Li',
]
if os.environ.get('USE_NCCL', '0') == '1': if os.environ.get('USE_NCCL', '0') == '1':
cxx_flags.append('-DFMOE_USE_NCCL') cxx_flags.append('-DFMOE_USE_NCCL')
ext_libs.append('nccl') ext_libs.append('nccl')
...@@ -14,9 +23,9 @@ if os.environ.get('USE_NCCL', '0') == '1': ...@@ -14,9 +23,9 @@ if os.environ.get('USE_NCCL', '0') == '1':
if __name__ == '__main__': if __name__ == '__main__':
setuptools.setup( setuptools.setup(
name='fastmoe', name='fastmoe',
version='0.1.2', version='0.2.0',
description='An efficient Mixture-of-Experts system for PyTorch', description='An efficient Mixture-of-Experts system for PyTorch',
author='Jiaao He, Jiezhong Qiu and Aohan Zeng', author=', '.join(authors),
author_email='hja20@mails.tsinghua.edu.cn', author_email='hja20@mails.tsinghua.edu.cn',
license='Apache-2', license='Apache-2',
url='https://github.com/laekov/fastmoe', url='https://github.com/laekov/fastmoe',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment