Commit 27b41933 authored by Alan Turner's avatar Alan Turner
Browse files

Add name field back to tune_ck

parent c8dda25a
...@@ -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)
......
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