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
7f15d11d
Commit
7f15d11d
authored
May 31, 2021
by
Rick Ho
Browse files
update version number
parent
d205aaeb
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
11 additions
and
2 deletions
+11
-2
setup.py
setup.py
+11
-2
No files found.
setup.py
View file @
7f15d11d
...
...
@@ -6,6 +6,15 @@ import os
cxx_flags
=
[]
ext_libs
=
[]
authors
=
[
'Jiaao He'
,
'Jiezhong Qiu'
,
'Aohan Zeng'
,
'Tiago Antunes'
,
'Jinjun Peng'
,
'Qin Li'
,
]
if
os
.
environ
.
get
(
'USE_NCCL'
,
'0'
)
==
'1'
:
cxx_flags
.
append
(
'-DFMOE_USE_NCCL'
)
ext_libs
.
append
(
'nccl'
)
...
...
@@ -14,9 +23,9 @@ if os.environ.get('USE_NCCL', '0') == '1':
if
__name__
==
'__main__'
:
setuptools
.
setup
(
name
=
'fastmoe'
,
version
=
'0.
1.2
'
,
version
=
'0.
2.0
'
,
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'
,
license
=
'Apache-2'
,
url
=
'https://github.com/laekov/fastmoe'
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment