You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by wu...@apache.org on 2022/04/26 17:17:51 UTC
[tvm] branch main updated (4dc47df369 -> 68464841ea)
This is an automated email from the ASF dual-hosted git repository.
wuwei pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 4dc47df369 allow constant value let binding in script (#11115)
add 68464841ea [Metaschedule] Auto tensorization for CPU / GPU dot product (#11088)
No new revisions were added by this update.
Summary of changes:
include/tvm/meta_schedule/postproc.h | 6 +-
include/tvm/meta_schedule/schedule_rule.h | 23 +
include/tvm/tir/stmt.h | 5 +
python/tvm/meta_schedule/postproc/__init__.py | 1 +
...e_cooperative_fetch.py => rewrite_tensorize.py} | 20 +-
python/tvm/meta_schedule/schedule_rule/__init__.py | 2 +-
.../schedule_rule/multi_level_tiling.py | 49 ++
python/tvm/meta_schedule/testing/te_workload.py | 2 +-
src/meta_schedule/postproc/rewrite_tensorize.cc | 105 +++++
.../schedule_rule/multi_level_tiling.cc | 25 +-
.../schedule_rule/multi_level_tiling.h | 30 ++
.../multi_level_tiling_with_intrin.cc | 79 ++++
.../test_meta_schedule_auto_tensorize.py | 347 ++++++++++++++
...est_meta_schedule_postproc_rewrite_tensorize.py | 509 +++++++++++++++++++++
...ta_schedule_schedule_rule_multi_level_tiling.py | 263 ++++++++++-
15 files changed, 1431 insertions(+), 35 deletions(-)
copy python/tvm/meta_schedule/postproc/{rewrite_cooperative_fetch.py => rewrite_tensorize.py} (63%)
create mode 100644 src/meta_schedule/postproc/rewrite_tensorize.cc
create mode 100644 src/meta_schedule/schedule_rule/multi_level_tiling_with_intrin.cc
create mode 100644 tests/python/integration/test_meta_schedule_auto_tensorize.py
create mode 100644 tests/python/unittest/test_meta_schedule_postproc_rewrite_tensorize.py