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
norm
vllm
Commits
aa50b17c
Commit
aa50b17c
authored
Apr 17, 2023
by
Woosuk Kwon
Browse files
Change plotting script
parent
0f4b3219
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
11 additions
and
2 deletions
+11
-2
plot/plot_normalized_latency.py
plot/plot_normalized_latency.py
+11
-2
No files found.
plot/plot_normalized_latency.py
View file @
aa50b17c
...
...
@@ -16,9 +16,9 @@ SYSTEMS = [
SYSTEM_TO_LABEL
=
{
'orca-constant'
:
'Orca (Max)'
,
'orca-power2'
:
'Orca (
Next power of
2)'
,
'orca-power2'
:
'Orca (
Pow
2)'
,
'orca-oracle'
:
'Orca (Oracle)'
,
'cacheflow'
:
'
Cache
Flow'
,
'cacheflow'
:
'
KV
Flow'
,
}
SYSTEM_TO_COLOR
=
{
...
...
@@ -178,6 +178,15 @@ def plot_normalized_latency(
else
:
plt
.
ylim
(
bottom
=
0
,
top
=
ylim
)
handles
,
labels
=
plt
.
gca
().
get_legend_handles_labels
()
handles
=
reversed
(
handles
)
labels
=
reversed
(
labels
)
plt
.
legend
(
handles
,
labels
,
ncol
=
4
,
fontsize
=
12
,
loc
=
'upper center'
,
bbox_to_anchor
=
(
0.5
,
1.15
),
columnspacing
=
0.5
,
handletextpad
=
0.5
,
handlelength
=
1.5
,
frameon
=
False
,
borderpad
=
0
)
# Save figure.
model
,
tp
=
get_model
(
exp_dir
)
sampling
=
get_sampling
(
exp_dir
)
...
...
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