__init__.py 794 Bytes
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
"""The auto-tuning module of tvm

This module includes:

* Tuning space definition API

* Efficient auto-tuners

* Tuning result and database support

* Distributed measurement to scale up tuning
"""

from . import database
from . import feature
from . import measure
from . import record
from . import task
from . import tuner
from . import util
21 22
from . import env
from . import tophub
23 24

# some shortcuts
25 26
from .measure import measure_option, MeasureInput, MeasureResult, MeasureErrorNo, \
    LocalBuilder, LocalRunner, RPCRunner
27
from .tuner import callback
28
from .task import template, get_config, create, ConfigSpace, ConfigEntity, \
29 30
    register_topi_compute, register_topi_schedule, \
    DispatchContext, FallbackContext, ApplyHistoryBest as apply_history_best
31
from .env import GLOBAL_SCOPE