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 28de742 [Refactor] Unify the shared pass prefix between vm and graph
(#8526)
add 4b67dac [CUDA] Support multiple TIR-level dynamic shared memory
allocations (#8571)
No new revisions were added by this update.
Summary of changes:
include/tvm/tir/transform.h | 5 +
python/tvm/driver/build_module.py | 5 +-
python/tvm/tir/transform/transform.py | 12 +
src/driver/driver_api.cc | 1 +
.../merge_dynamic_shared_memory_allocations.cc | 149 ++++++++++++
...form_merge_dynamic_shared_memory_allocations.py | 259 +++++++++++++++++++++
6 files changed, 430 insertions(+), 1 deletion(-)
create mode 100644
src/tir/transforms/merge_dynamic_shared_memory_allocations.cc
create mode 100644
tests/python/unittest/test_tir_transform_merge_dynamic_shared_memory_allocations.py
