File tree Expand file tree Collapse file tree 2 files changed +6
-5
lines changed
Expand file tree Collapse file tree 2 files changed +6
-5
lines changed Original file line number Diff line number Diff line change 2424def 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
3334schema_v1_0 = {
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments