Skip to content

GitLab

Commit 424b58b4 authored by Paul's avatar Paul
Browse files

Catch exception

parent 19bf60bd
Hide whitespace changes
Inline Side-by-side
Showing with 17 additions and 15 deletions
tools/tune_ck.py
View file @ 424b58b4
...@@ -44,22 +44,24 @@ def get_device_time(s): ...@@ -44,22 +44,24 @@ def get_device_time(s):
def benchmark_ck(config, tuning): def benchmark_ck(config, tuning):
b = { try:
'settings': { b = {
'iterations': 100 'settings': {
}, 'iterations': 100
'compile_op': { },
'name': 'ck_gemm', 'compile_op': {
'check': True, 'name': 'ck_gemm',
'tuning_val': tuning, 'check': True,
'inputs': config 'tuning_val': tuning,
'inputs': config
}
} }
} for line in run_driver(b):
for line in run_driver(b): dtime = get_device_time(line)
dtime = get_device_time(line) print(dtime)
print(dtime) return dtime
return dtime finally:
return sys.float_info.max return sys.float_info.max
def benchmark(config, size): def benchmark(config, size):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment