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/tvm.git.
from 783be9d [TOPI] disable test_shift with i8 datatype (#7597)
add 0b4f669 [AutoSchedule] Sparse dense tuning support with custom sketch
rule (#7313)
No new revisions were added by this update.
Summary of changes:
include/tvm/auto_scheduler/measure_record.h | 2 +-
include/tvm/auto_scheduler/search_task.h | 8 +-
python/tvm/auto_scheduler/__init__.py | 1 +
python/tvm/auto_scheduler/measure.py | 166 +++++++++-
python/tvm/auto_scheduler/search_task.py | 191 +++++++++++-
python/tvm/auto_scheduler/utils.py | 3 +
python/tvm/topi/nn/sparse.py | 118 ++++++-
src/auto_scheduler/feature.cc | 9 +-
src/auto_scheduler/measure_record.cc | 34 +++
src/auto_scheduler/search_policy/utils.cc | 16 +
src/auto_scheduler/search_task.cc | 7 +-
.../python/unittest/test_auto_scheduler_measure.py | 32 +-
.../unittest/test_auto_scheduler_search_task.py | 207 +++++++++++++
tutorials/auto_scheduler/ci_logs/sparse_dense.json | 2 +
tutorials/auto_scheduler/tune_sparse_x86.py | 339 +++++++++++++++++++++
15 files changed, 1109 insertions(+), 26 deletions(-)
create mode 100644 tests/python/unittest/test_auto_scheduler_search_task.py
create mode 100644 tutorials/auto_scheduler/ci_logs/sparse_dense.json
create mode 100644 tutorials/auto_scheduler/tune_sparse_x86.py
