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
zhaoyu6
sglang
Commits
2f2e0743
"tests/unit/profiling/vscode:/vscode.git/clone" did not exist on "c25a91b60c5192065dfdcabd373b947aa2234fe1"
Unverified
Commit
2f2e0743
authored
Nov 16, 2024
by
Lianmin Zheng
Committed by
GitHub
Nov 16, 2024
Browse files
Fix weight update for data parallelism (#2050)
parent
2ffe0a73
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
6 additions
and
3 deletions
+6
-3
python/sglang/bench_offline_throughput.py
python/sglang/bench_offline_throughput.py
+5
-3
python/sglang/srt/managers/data_parallel_controller.py
python/sglang/srt/managers/data_parallel_controller.py
+1
-0
No files found.
python/sglang/bench_offline_throughput.py
View file @
2f2e0743
...
@@ -231,15 +231,16 @@ def throughput_test(
...
@@ -231,15 +231,16 @@ def throughput_test(
input_requests
=
get_dataset
(
bench_args
,
tokenizer
)
input_requests
=
get_dataset
(
bench_args
,
tokenizer
)
warmup_requests
=
sample_random_requests
(
warmup_requests
=
sample_random_requests
(
input_len
=
2
0
,
input_len
=
2
56
,
output_len
=
4
,
output_len
=
16
,
num_prompts
=
2
,
num_prompts
=
16
,
range_ratio
=
0.8
,
range_ratio
=
0.8
,
tokenizer
=
tokenizer
,
tokenizer
=
tokenizer
,
dataset_path
=
bench_args
.
dataset_path
,
dataset_path
=
bench_args
.
dataset_path
,
)
)
# Warm up
# Warm up
logging
.
info
(
"
\n
Warmup..."
)
throughput_test_once
(
throughput_test_once
(
backend_name
=
bench_args
.
backend
,
backend_name
=
bench_args
.
backend
,
backend
=
backend
,
backend
=
backend
,
...
@@ -247,6 +248,7 @@ def throughput_test(
...
@@ -247,6 +248,7 @@ def throughput_test(
ignore_eos
=
not
bench_args
.
disable_ignore_eos
,
ignore_eos
=
not
bench_args
.
disable_ignore_eos
,
)
)
logging
.
info
(
"
\n
Benchmark..."
)
result
=
throughput_test_once
(
result
=
throughput_test_once
(
backend_name
=
bench_args
.
backend
,
backend_name
=
bench_args
.
backend
,
backend
=
backend
,
backend
=
backend
,
...
...
python/sglang/srt/managers/data_parallel_controller.py
View file @
2f2e0743
...
@@ -83,6 +83,7 @@ class DataParallelController:
...
@@ -83,6 +83,7 @@ class DataParallelController:
self
.
workers
=
[]
self
.
workers
=
[]
for
dp_rank
in
range
(
server_args
.
dp_size
):
for
dp_rank
in
range
(
server_args
.
dp_size
):
tmp_port_args
=
PortArgs
.
init_new
(
server_args
)
tmp_port_args
=
PortArgs
.
init_new
(
server_args
)
tmp_port_args
.
tokenizer_ipc_name
=
port_args
.
tokenizer_ipc_name
tmp_port_args
.
detokenizer_ipc_name
=
port_args
.
detokenizer_ipc_name
tmp_port_args
.
detokenizer_ipc_name
=
port_args
.
detokenizer_ipc_name
send_to
=
self
.
launch_tensor_parallel_group
(
send_to
=
self
.
launch_tensor_parallel_group
(
...
...
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