You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by gi...@apache.org on 2022/07/14 23:06:11 UTC
[tvm] branch last-successful updated (9a16503443 -> e084791852)
This is an automated email from the ASF dual-hosted git repository.
github-bot pushed a change to branch last-successful
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 9a16503443 adding Alexey (#12090)
add e084791852 [MetaSchedule] Add MultiLevelTilingTensorCore rule for auto-tensorization on CUDA (#12059)
No new revisions were added by this update.
Summary of changes:
include/tvm/meta_schedule/schedule_rule.h | 24 ++
include/tvm/tir/stmt.h | 4 +
python/tvm/meta_schedule/schedule_rule/__init__.py | 7 +-
.../schedule_rule/multi_level_tiling.py | 54 ++-
python/tvm/meta_schedule/testing/schedule_rule.py | 34 ++
python/tvm/tir/tensor_intrin/cuda.py | 73 +++-
.../postproc/rewrite_reduction_block.cc | 17 +
src/meta_schedule/postproc/rewrite_tensorize.cc | 6 +-
src/meta_schedule/schedule_rule/auto_inline.cc | 3 +-
.../schedule_rule/multi_level_tiling.cc | 28 +-
.../schedule_rule/multi_level_tiling.h | 10 +-
.../multi_level_tiling_tensor_core.cc | 393 +++++++++++++++++++
src/tir/schedule/analysis.h | 18 +-
src/tir/schedule/analysis/analysis.cc | 33 +-
src/tir/schedule/primitive/block_annotate.cc | 5 +-
src/tir/schedule/primitive/cache_read_write.cc | 7 +-
.../schedule/primitive/layout_transformation.cc | 7 +-
src/tir/schedule/transform.cc | 6 +-
...est_meta_schedule_postproc_rewrite_tensorize.py | 2 +-
...ta_schedule_schedule_rule_multi_level_tiling.py | 426 ++++++++++++++++++++-
20 files changed, 1092 insertions(+), 65 deletions(-)
create mode 100644 src/meta_schedule/schedule_rule/multi_level_tiling_tensor_core.cc