Skip to content

Commit e20cb54

Browse files
add default optimization direction for 'fitness' and 'cost'
As we extend to more optimization objectives and explore constrained optimization it happens more often that we have a user-defined metric that defines the optimization objective. This pull makes it easy to specify the optimization direction, without the need to explicitly specify the direction using the ``objective_higher_is_better=`` option of ``tune_kernel()``. Now, if your objective is named ``"fitness"`` Kernel Tuner assumes maximization and for objectives named ``"cost"`` or ``"loss"`` minimization is assumed.
1 parent a48abc1 commit e20cb54

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

kernel_tuner/integration.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,9 @@
1111
objective_default_map = {
1212
"time": False,
1313
"energy": False,
14+
"fitness": True,
15+
"cost": False,
16+
"loss": False,
1417
"GFLOP/s": True,
1518
"TFLOP/s": True,
1619
"GB/s": True,

0 commit comments

Comments
 (0)