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