You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by lm...@apache.org on 2020/10/30 23:59:27 UTC
[incubator-tvm] branch main updated (7196eb8 -> a261454)
This is an automated email from the ASF dual-hosted git repository.
lmzheng pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git.
from 7196eb8 [BYOC] Allow custom codegens to register their own constant updater (#6697)
add a261454 [AutoScheduler] Relay integration : Task extraction (#6710)
No new revisions were added by this update.
Summary of changes:
python/tvm/auto_scheduler/__init__.py | 6 +
python/tvm/auto_scheduler/compute_dag.py | 9 +-
python/tvm/auto_scheduler/cost_model/xgb_model.py | 22 +-
python/tvm/auto_scheduler/dispatcher.py | 275 +++++++++++++++++++++
.../generic/sort.py => auto_scheduler/env.py} | 48 ++--
python/tvm/auto_scheduler/measure.py | 118 +++++----
python/tvm/auto_scheduler/relay_integration.py | 232 +++++++++++++++++
python/tvm/auto_scheduler/search_policy.py | 11 +-
python/tvm/auto_scheduler/utils.py | 5 +-
python/tvm/auto_scheduler/workload_registry.py | 119 +++++----
python/tvm/relay/backend/compile_engine.py | 27 +-
python/tvm/relay/op/op.py | 30 ++-
python/tvm/relay/op/strategy/cuda.py | 13 +-
python/tvm/relay/op/strategy/generic.py | 2 +-
src/auto_scheduler/search_policy/sketch_policy.cc | 30 ++-
src/auto_scheduler/search_policy/sketch_policy.h | 7 +-
src/relay/backend/compile_engine.cc | 24 +-
.../relay/test_auto_scheduler_task_extraction.py | 90 +++++++
tests/python/relay/test_auto_scheduler_tuning.py | 62 +++++
.../unittest/test_auto_scheduler_cost_model.py | 2 +-
.../test_auto_scheduler_evolutionary_search.py | 4 +-
21 files changed, 959 insertions(+), 177 deletions(-)
create mode 100644 python/tvm/auto_scheduler/dispatcher.py
copy python/tvm/{topi/generic/sort.py => auto_scheduler/env.py} (51%)
create mode 100644 python/tvm/auto_scheduler/relay_integration.py
create mode 100644 tests/python/relay/test_auto_scheduler_task_extraction.py
create mode 100644 tests/python/relay/test_auto_scheduler_tuning.py