Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
a3dec59c
Commit
a3dec59c
authored
Feb 12, 2025
by
王敏
Browse files
优化benchmark moe
parent
44d451ff
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
benchmarks/kernels/benchmark_moe.py
benchmarks/kernels/benchmark_moe.py
+1
-1
No files found.
benchmarks/kernels/benchmark_moe.py
View file @
a3dec59c
...
...
@@ -641,7 +641,7 @@ if __name__ == "__main__":
parser
.
add_argument
(
"--seed"
,
type
=
int
,
default
=
0
)
parser
.
add_argument
(
"--batch-size"
,
type
=
int
,
required
=
False
)
parser
.
add_argument
(
"--tune"
,
action
=
"store_true"
)
parser
.
add_argument
(
"--nn
_
moe"
,
type
=
bool
,
default
=
Tru
e
)
parser
.
add_argument
(
"--nn
-
moe"
,
action
=
'store_true'
,
default
=
Fals
e
)
parser
.
add_argument
(
"--trust-remote-code"
,
action
=
"store_true"
)
parser
.
add_argument
(
"--moe-ep-size"
,
type
=
int
,
default
=
1
)
parser
.
add_argument
(
"--num-gpus"
,
type
=
int
,
default
=
1
)
...
...
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