This is an automated email from the ASF dual-hosted git repository.
andrewzhaoluo pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 3f56a95b87 [TVMScript] Use new variable frame in If/Then/Else (#14250)
add e5ae4347dd [CUDA][Schedule] Better Layout Transform Schedules (#14167)
No new revisions were added by this update.
Summary of changes:
include/tvm/topi/transform.h | 10 +-
python/tvm/meta_schedule/schedule/cuda/__init__.py | 2 +
.../schedule/cuda/layout_transform.py | 583 +++++++++++++++++++++
python/tvm/relay/op/_transform.py | 2 +-
python/tvm/relay/op/strategy/cuda.py | 11 +
python/tvm/relay/op/strategy/generic.py | 36 +-
python/tvm/topi/transform.py | 17 +-
src/topi/transform.cc | 2 +-
.../test_meta_schedule_relay_integration.py | 3 +
...meta_schedule_schedule_cuda_layout_transform.py | 466 ++++++++++++++++
10 files changed, 1116 insertions(+), 16 deletions(-)
create mode 100644 python/tvm/meta_schedule/schedule/cuda/layout_transform.py
create mode 100644
tests/python/unittest/test_meta_schedule_schedule_cuda_layout_transform.py
