How to define search space an bypass some of them

Hi guys:

Do we have some functions which I can use to bypass some configurations in pre-defined search space.

For instance, If I I have three parameters should be tuned p1 ,p2 , p2

Followed and auto-tuning tutorial, I should do something like

cfg.define_knob("p1", [1, 2, 4])
cfg.define_knob("p2", [1, 2, 4])
cfg.define_knob("p3", [1, 2, 4])

In this case, we will have 3 x 3 x 3 = 27 configurations.

However, If I know that I will certainly not go to p1 = 1 , p2 = 2 , p3 =3 because some known exteran hardware information. Then I can prune some configurations and reduce the 27-configuration search space to smaller one.

Do we have some auxiliary functions for this? @merrymercy @tqchen Thank you

We don’t support non-grid tuning space in AutoTVM. On the other hand, the new auto-scheduler that is being upstreamed will support symbolic tuning parameters with conditional expression.

1 Like