knobs

KNOBS: tuple[KnobSpec, ...] = (ACCUMULATION, KERNEL)

KnobSpec(
    name: str,
    values: tuple[str, ...],
    default: str,
    approximate_values: frozenset[str],
    auto: Mapping[OptimizationGoal, str],
    applies_to: frozenset[str],
    summary: str,
) -> None

dataclass

One optimization knob.

name: str

The knob’s configuration key under optimization and operators[].optimization.

values: tuple[str, ...]

Its values, excluding auto.

default: str

The value that keeps the kernel library’s default numerics; auto falls back to it when the approximation gate is closed.

approximate_values: frozenset[str]

Values that change numerics relative to default.

auto: Mapping[OptimizationGoal, str]

The value auto picks for each goal, subject to the approximation gate.

applies_to: frozenset[str]

Operator type keys the knob can apply to.

summary: str

What the knob changes.

resolve(requested: str, goal: OptimizationGoal, allow_approximate: bool) -> str

Resolve a requested value; auto follows the static table.

An explicit value is used as given. auto takes the goal’s value, unless that value is approximate and the gate is closed, in which case it takes default.

Parameters

Parameters of resolve
Name	Type	Default	Description
`requested`	`str`	Required	The requested value, possibly ``auto``.
`goal`	`OptimizationGoal`	Required	``optimization.goal``.
`allow_approximate`	`bool`	Required	``optimization.allow_approximate``.

Returns

Returns of resolve
Value	Type	Description
`str`	`str`	A value from ``values``.

Raises

Errors raised by resolve
Type	Description
`ValueError`	If ``requested`` is neither ``auto`` nor one of ``values``.

is_approximate(value: str) -> bool

Whether value changes numerics relative to default.

knob_names() -> frozenset[str]

The names of all knobs.

knob(name: str) -> KnobSpec

Look up a knob by name.

Parameters

Parameters of knob
Name	Type	Default	Description
`name`	`str`	Required	The knob's name.

Returns

Returns of knob
Value	Type	Description
`KnobSpec`	`KnobSpec`	The knob.

Raises

Errors raised by knob
Type	Description
`KeyError`	When no knob has that name.

default_knob_values() -> dict[str, str]

Each knob’s default value, the choice an operator makes without a handler.