This is an automated email from the ASF dual-hosted git repository.
junrushao pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from 3181f27 skip aot checks when USE_MICRO=OFF (#8772)
add da4869e [TensorIR][M2a] Parallel, Vectorize, Bind & Unroll (#8716)
No new revisions were added by this update.
Summary of changes:
include/tvm/tir/schedule/schedule.h | 39 ++-
include/tvm/tir/transform.h | 12 +
python/tvm/tir/schedule/schedule.py | 240 +++++++++++++-
python/tvm/tir/transform/transform.py | 22 ++
src/driver/driver_api.cc | 1 +
src/runtime/thread_storage_scope.h | 2 +-
src/tir/schedule/analysis.h | 23 ++
src/tir/schedule/analysis/analysis.cc | 59 ++++
src/tir/schedule/concrete_schedule.cc | 54 ++-
src/tir/schedule/concrete_schedule.h | 6 +-
src/tir/schedule/primitive.h | 43 ++-
src/tir/schedule/primitive/for_kind.cc | 289 ++++++++++++++++
src/tir/schedule/primitive/reduction.cc | 4 +-
src/tir/schedule/schedule.cc | 6 +
src/tir/schedule/state.cc | 8 +-
src/tir/schedule/traced_schedule.cc | 40 +++
src/tir/schedule/traced_schedule.h | 4 +
src/tir/transforms/flatten_buffer.cc | 5 +-
src/tir/transforms/inject_virtual_thread.cc | 11 +-
src/tir/transforms/unify_thread_binding.cc | 144 ++++++++
.../python/unittest/test_tir_schedule_for_kind.py | 365 +++++++++++++++++++++
.../python/unittest/test_tir_schedule_reduction.py | 30 +-
.../test_tir_transform_compact_buffer_region.py | 1 +
.../test_tir_transform_convert_blocks_to_opaque.py | 1 +
.../unittest/test_tir_transform_flatten_buffer.py | 1 +
.../test_tir_transform_lower_init_block.py | 1 +
...sform_plan_update_buffer_allocation_location.py | 1 +
.../unittest/test_tir_transform_storage_flatten.py | 1 +
.../test_tir_transform_unify_thread_binding.py | 227 +++++++++++++
29 files changed, 1600 insertions(+), 40 deletions(-)
create mode 100644 src/tir/schedule/primitive/for_kind.cc
create mode 100644 src/tir/transforms/unify_thread_binding.cc
create mode 100644 tests/python/unittest/test_tir_schedule_for_kind.py
create mode 100644
tests/python/unittest/test_tir_transform_unify_thread_binding.py
