This is an automated email from the ASF dual-hosted git repository.
zhaowu pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git.
from f34e3a8 [RUNTIME][FFI] Fix cython FFI compact with np.int64 (#6321)
add b1f8f15 [Ansor][AutoTVM v2.0] Phase 2: Basic GPU Sketch Search Policy
(#6269)
No new revisions were added by this update.
Summary of changes:
include/tvm/auto_scheduler/search_policy.h | 35 ++
python/tvm/auto_scheduler/__init__.py | 3 +-
python/tvm/auto_scheduler/auto_schedule.py | 122 +------
python/tvm/auto_scheduler/loop_state.py | 9 +-
python/tvm/auto_scheduler/search_policy.py | 180 ++++++++++
src/auto_scheduler/auto_schedule.cc | 13 +-
src/auto_scheduler/search_policy/search_policy.cc | 63 ++++
src/auto_scheduler/search_policy/sketch_policy.cc | 72 +++-
src/auto_scheduler/search_policy/sketch_policy.h | 2 +
.../search_policy/sketch_policy_rules.cc | 378 +++++++++++++++++++--
.../search_policy/sketch_policy_rules.h | 123 +++----
src/auto_scheduler/search_policy/utils.cc | 55 +++
src/auto_scheduler/search_policy/utils.h | 184 ++++++++++
src/auto_scheduler/search_task.cc | 28 +-
src/auto_scheduler/transform_step.cc | 100 +++---
src/auto_scheduler/utils.h | 21 --
.../unittest/test_auto_scheduler_search_policy.py | 29 +-
.../test_auto_scheduler_sketch_generation.py | 286 +++++++++++++++-
18 files changed, 1387 insertions(+), 316 deletions(-)
create mode 100644 python/tvm/auto_scheduler/search_policy.py
