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
gaoqiong
MIGraphX
Commits
27b41933
Commit
27b41933
authored
Mar 31, 2023
by
Alan Turner
Browse files
Add name field back to tune_ck
parent
c8dda25a
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
4 deletions
+4
-4
tools/tune_ck.py
tools/tune_ck.py
+4
-4
No files found.
tools/tune_ck.py
View file @
27b41933
...
@@ -49,13 +49,13 @@ def get_device_time(s):
...
@@ -49,13 +49,13 @@ def get_device_time(s):
return
convert_to_float
(
fields
[
-
1
].
strip
())
return
convert_to_float
(
fields
[
-
1
].
strip
())
def
run_driver_ck
(
config
,
tuning
,
iterations
):
def
run_driver_ck
(
config
,
name
,
tuning
,
iterations
):
b
=
{
b
=
{
'settings'
:
{
'settings'
:
{
'iterations'
:
iterations
'iterations'
:
iterations
},
},
'compile_op'
:
{
'compile_op'
:
{
'name'
:
'ck_gemm'
,
'name'
:
name
,
'check'
:
True
,
'check'
:
True
,
'tuning_val'
:
tuning
,
'tuning_val'
:
tuning
,
'inputs'
:
config
'inputs'
:
config
...
@@ -64,9 +64,9 @@ def run_driver_ck(config, tuning, iterations):
...
@@ -64,9 +64,9 @@ def run_driver_ck(config, tuning, iterations):
return
run_driver
(
b
)
return
run_driver
(
b
)
def
benchmark_ck
(
config
,
tuning
):
def
benchmark_ck
(
config
,
name
,
tuning
):
try
:
try
:
for
line
in
run_driver_ck
(
config
,
tuning
,
100
):
for
line
in
run_driver_ck
(
config
,
name
,
tuning
,
100
):
dtime
=
get_device_time
(
line
)
dtime
=
get_device_time
(
line
)
print
(
dtime
)
print
(
dtime
)
return
float
(
dtime
)
return
float
(
dtime
)
...
...
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