[Ansor][AutoTVM v2.0] Phase 2: Basic GPU Sketch Search Policy (#6269)
authorChenfan <chengfan.jcf@alibaba-inc.com>
Mon, 24 Aug 2020 10:46:31 +0000 (18:46 +0800)
committerGitHub <noreply@github.com>
Mon, 24 Aug 2020 10:46:31 +0000 (18:46 +0800)
commitb1f8f15031848f55855cc8489cba3d8faeb65b3f
tree12c667e9288ddd0148efedd226b92d6e87a2eeb9
parentf34e3a88a01a347e7cee02c7085b837baef1fb72
[Ansor][AutoTVM v2.0] Phase 2: Basic GPU Sketch Search Policy (#6269)

* Add PreloadMeasuredStates & Split search_policy.py

* Add GPU sketch rule

* Update

* Bug fix for log record

* Lint fix

* Update tutorial

* Update

* UT fix

* Remove tutorial

* Update

* Update

* Update UT

* Lint fix

* Update

* Update
18 files changed:
include/tvm/auto_scheduler/search_policy.h
python/tvm/auto_scheduler/__init__.py
python/tvm/auto_scheduler/auto_schedule.py
python/tvm/auto_scheduler/loop_state.py
python/tvm/auto_scheduler/search_policy.py [new file with mode: 0644]
src/auto_scheduler/auto_schedule.cc
src/auto_scheduler/search_policy/search_policy.cc
src/auto_scheduler/search_policy/sketch_policy.cc
src/auto_scheduler/search_policy/sketch_policy.h
src/auto_scheduler/search_policy/sketch_policy_rules.cc
src/auto_scheduler/search_policy/sketch_policy_rules.h
src/auto_scheduler/search_policy/utils.cc
src/auto_scheduler/search_policy/utils.h
src/auto_scheduler/search_task.cc
src/auto_scheduler/transform_step.cc
src/auto_scheduler/utils.h
tests/python/unittest/test_auto_scheduler_search_policy.py
tests/python/unittest/test_auto_scheduler_sketch_generation.py