[Ansor][AutoTVM v2.0] Phase 1: Add cache_read/cache_write steps (#6107)
authorChenfan <chengfan.jcf@alibaba-inc.com>
Mon, 27 Jul 2020 04:35:52 +0000 (12:35 +0800)
committerGitHub <noreply@github.com>
Mon, 27 Jul 2020 04:35:52 +0000 (21:35 -0700)
commitb8f8b8d9a804f5867daddd9c5d7ddc0ed4d0d199
tree5c1acca796100a8283683f69b9dcf2fb348d6949
parent959cff1c786e0eb33b99007be66de61d2275d7a5
[Ansor][AutoTVM v2.0] Phase 1: Add cache_read/cache_write steps (#6107)

* Add cache_read/cache_write step

* Update

* Update

* Update

* Update state->current_compute_dag to Optional

* Update

* Update doc

* Update

* Update

* Doc update

* Update
include/tvm/auto_scheduler/compute_dag.h
include/tvm/auto_scheduler/loop_state.h
include/tvm/auto_scheduler/transform_step.h
python/tvm/auto_scheduler/compute_dag.py
python/tvm/auto_scheduler/loop_state.py
src/auto_scheduler/compute_dag.cc
src/auto_scheduler/loop_state.cc
src/auto_scheduler/transform_step.cc
tests/python/unittest/test_auto_scheduler_loop_state.py
tests/python/unittest/test_auto_scheduler_measure.py