Skip to content

Commit 7797dbc

Browse files
bugfix get_objective_defaults, issue #204
1 parent 324aee9 commit 7797dbc

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

kernel_tuner/integration.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,10 +24,11 @@
2424
def get_objective_defaults(objective, objective_higher_is_better):
2525
""" Uses time as default objective and attempts to lookup objective_higher_is_better for known objectives """
2626
objective = objective or "time"
27-
if objective_higher_is_better is None and objective in objective_default_map:
28-
objective_higher_is_better = objective_default_map[objective]
29-
else:
30-
raise ValueError(f"Please specify objective_higher_is_better for objective {objective}")
27+
if objective_higher_is_better is None:
28+
if objective in objective_default_map:
29+
objective_higher_is_better = objective_default_map[objective]
30+
else:
31+
raise ValueError(f"Please specify objective_higher_is_better for objective {objective}")
3132
return objective, objective_higher_is_better
3233

3334
schema_v1_0 = {

test/test_runners.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ def test_sequential_runner_alt_block_size_names(env):
6767
result, _ = tune_kernel(*env,
6868
grid_div_x=["block_dim_x"],
6969
answer=answer,
70-
block_size_names=block_size_names)
70+
block_size_names=block_size_names, objective='time', objective_higher_is_better=False)
7171

7272
assert len(result) == len(tune_params["block_dim_x"])
7373

0 commit comments

Comments
 (0)