This is an automated email from the ASF dual-hosted git repository.
lmzheng pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git.
from db6e0c1 [JVM] Support overriding RPCWatchdog termination behavior on
Android and other platforms (#6216)
add 75b8318 [Ansor][AutoTVM v2.0] Phase 2: Basic CPU Sketch Search Policy
(#6184)
No new revisions were added by this update.
Summary of changes:
include/tvm/auto_scheduler/auto_schedule.h | 16 +-
include/tvm/auto_scheduler/compute_dag.h | 18 +-
include/tvm/auto_scheduler/cost_model.h | 12 +-
include/tvm/auto_scheduler/search_policy.h | 37 +-
include/tvm/auto_scheduler/transform_step.h | 2 +-
python/tvm/auto_scheduler/__init__.py | 2 +-
python/tvm/auto_scheduler/auto_schedule.py | 126 ++++-
src/auto_scheduler/auto_schedule.cc | 24 +-
src/auto_scheduler/compute_dag.cc | 94 +++-
src/auto_scheduler/cost_model.cc | 14 +-
src/auto_scheduler/loop_state.cc | 8 +-
src/auto_scheduler/search_policy/empty_policy.cc | 35 +-
src/auto_scheduler/search_policy/empty_policy.h | 7 +-
src/auto_scheduler/search_policy/search_policy.cc | 14 +-
src/auto_scheduler/search_policy/sketch_policy.cc | 401 ++++++++++++++
src/auto_scheduler/search_policy/sketch_policy.h | 176 +++++++
.../search_policy/sketch_policy_rules.cc | 584 +++++++++++++++++++++
.../search_policy/sketch_policy_rules.h | 207 ++++++++
src/auto_scheduler/search_policy/utils.cc | 286 ++++++++++
src/auto_scheduler/search_policy/utils.h | 484 +++++++++++++++++
src/auto_scheduler/utils.h | 47 +-
tests/cpp/auto_scheduler_test.cc | 4 +-
.../python/unittest/test_auto_scheduler_common.py | 112 +++-
.../unittest/test_auto_scheduler_loop_state.py | 7 +-
.../unittest/test_auto_scheduler_search_policy.py | 31 +-
.../test_auto_scheduler_sketch_generation.py | 102 ++++
26 files changed, 2713 insertions(+), 137 deletions(-)
create mode 100644 src/auto_scheduler/search_policy/sketch_policy.cc
create mode 100644 src/auto_scheduler/search_policy/sketch_policy.h
create mode 100644 src/auto_scheduler/search_policy/sketch_policy_rules.cc
create mode 100644 src/auto_scheduler/search_policy/sketch_policy_rules.h
create mode 100644 src/auto_scheduler/search_policy/utils.cc
create mode 100644 src/auto_scheduler/search_policy/utils.h
create mode 100644
tests/python/unittest/test_auto_scheduler_sketch_generation.py
