This is an automated email from the ASF dual-hosted git repository.
github-bot pushed a change to branch nightly
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 969fad363b [TIR] Add VisitBufferDef/VisitBufferUse to base
StmtVisitor/StmtMutator (#18873)
add 079e4af391 [REFACTOR][TIR] Rename LetStmt to Bind and flatten to
sequential semantics (#18874)
add dd7e260656 [chore] Update docker/README.md documentation and fix links
(#18875)
add 419a8c861e Batched GPU dispatch and object caching for WebGPU runtime
(#18871)
add 14e41c681a [Relax][ONNX] Support dynamic repeats for Tile (#18878)
No new revisions were added by this update.
Summary of changes:
docker/README.md | 8 +-
include/tvm/script/ir_builder/tir/frame.h | 42 ---
include/tvm/script/ir_builder/tir/ir.h | 13 +-
include/tvm/script/printer/ir_docsifier_functor.h | 1 +
include/tvm/tir/stmt.h | 36 +-
include/tvm/tir/stmt_functor.h | 8 +-
include/tvm/tir/var.h | 2 +-
python/tvm/relax/frontend/onnx/onnx_frontend.py | 49 ++-
python/tvm/script/ir_builder/tir/frame.py | 7 -
python/tvm/script/ir_builder/tir/ir.py | 30 +-
python/tvm/script/parser/core/parser.py | 3 +
python/tvm/script/parser/tir/parser.py | 14 +-
python/tvm/tir/__init__.py | 2 +-
python/tvm/tir/functor.py | 36 +-
python/tvm/tir/stmt.py | 22 +-
python/tvm/tir/transform/transform.py | 6 +-
python/tvm/topi/transform.py | 22 ++
src/arith/ir_mutator_with_analyzer.cc | 8 +-
src/arith/ir_mutator_with_analyzer.h | 2 +-
src/arith/ir_visitor_with_analyzer.cc | 3 +-
src/arith/ir_visitor_with_analyzer.h | 2 +-
src/relax/op/tensor/inspect.cc | 51 ++-
src/s_tir/analysis/estimate_flops.cc | 3 +-
.../analysis/sblock_access_region_detector.cc | 5 +-
src/s_tir/backend/adreno/inject_texture_alloc.cc | 6 +-
src/s_tir/schedule/analysis/reducer.cc | 59 ++--
src/s_tir/schedule/primitive/blockize_tensorize.cc | 1 +
.../schedule/primitive/layout_transformation.cc | 36 +-
src/s_tir/schedule/primitive/reduction.cc | 13 +-
src/s_tir/transform/compact_buffer_region.cc | 6 +-
src/s_tir/transform/hoist_expression.cc | 64 ++--
src/s_tir/transform/inject_virtual_thread.cc | 63 +++-
src/s_tir/transform/lower_thread_allreduce.cc | 8 +-
src/s_tir/transform/lower_vtcm_alloc.cc | 6 +-
src/s_tir/transform/profile_instrumentation.cc | 14 +-
src/s_tir/transform/remove_store_undef.cc | 65 +++-
src/s_tir/transform/renew_defs.cc | 2 +-
src/s_tir/transform/storage_access.cc | 4 +-
src/s_tir/transform/storage_access.h | 2 +-
src/script/ir_builder/tir/frame.cc | 6 -
src/script/ir_builder/tir/ir.cc | 33 +-
src/script/printer/relax/distributed.cc | 1 +
src/script/printer/tir/stmt.cc | 31 +-
src/target/llvm/codegen_llvm.cc | 3 +-
src/target/llvm/codegen_llvm.h | 2 +-
src/target/source/codegen_c.cc | 4 +-
src/target/source/codegen_c.h | 2 +-
src/target/source/codegen_webgpu.cc | 3 +-
src/target/source/codegen_webgpu.h | 2 +-
src/target/spirv/codegen_spirv.cc | 3 +-
src/target/spirv/codegen_spirv.h | 2 +-
src/te/operation/create_primfunc.cc | 13 +-
src/tir/analysis/control_flow_graph.cc | 17 +-
src/tir/analysis/var_use_def_analysis.cc | 2 +-
src/tir/analysis/var_use_def_analysis.h | 2 +-
src/tir/analysis/verify_memory.cc | 2 +-
src/tir/analysis/verify_ssa.cc | 2 +-
src/tir/ir/data_type_rewriter.cc | 21 +-
src/tir/ir/data_type_rewriter.h | 4 +-
src/tir/ir/py_functor.cc | 26 +-
src/tir/ir/specialize.cc | 4 +-
src/tir/ir/stmt.cc | 19 +-
src/tir/ir/stmt_functor.cc | 13 +-
src/tir/ir/tir_visitor_with_path.cc | 30 +-
src/tir/ir/tir_visitor_with_path.h | 11 +-
src/tir/transform/common_subexpr_elim.cc | 326 ++++++++++++++----
src/tir/transform/common_subexpr_elim.h | 84 ++++-
src/tir/transform/ir_utils.cc | 345 ++++++++++++-------
src/tir/transform/ir_utils.h | 4 +-
src/tir/transform/lower_tvm_builtin.cc | 172 +++++++---
src/tir/transform/remove_no_op.cc | 20 +-
src/tir/transform/simplify.cc | 107 +-----
src/tir/transform/split_host_device.cc | 4 +-
src/tir/transform/storage_rewrite.cc | 21 +-
src/tir/transform/tvm_ffi_binder.cc | 18 +-
src/tir/transform/tvm_ffi_binder.h | 12 +-
src/tir/transform/unsupported_dtype_legalize.cc | 14 +-
src/tir/transform/vectorize_loop.cc | 12 +-
src/topi/transform.cc | 5 +
tests/python/relax/test_frontend_onnx.py | 31 ++
.../s_tir/schedule/test_tir_schedule_rfactor.py | 12 +-
.../test_s_tir_transform_hoist_expression.py | 23 +-
.../test_s_tir_transform_inject_ptx_async_copy.py | 8 +-
...sform_plan_update_buffer_allocation_location.py | 2 +-
.../transform/test_s_tir_transform_thread_sync.py | 36 +-
.../tir-analysis/test_tir_analysis_verify_ssa.py | 4 +-
.../test_tir_analysis_verify_well_formed.py | 92 ++++--
tests/python/tir-base/test_tir_constructor.py | 5 +-
tests/python/tir-base/test_tir_nodes.py | 4 +-
.../tir-base/test_tir_structural_equal_hash.py | 2 +-
.../test_tir_inline_private_functions.py | 30 +-
.../test_tir_transform_common_subexpr_elim.py | 364 ++++++++++-----------
.../test_tir_transform_convert_ssa.py | 81 +++--
.../test_tir_transform_lower_tvm_builtin.py | 4 +-
.../test_tir_transform_prim_func_pass.py | 2 +-
.../test_tir_transform_remove_no_op.py | 28 +-
.../tir-transform/test_tir_transform_simplify.py | 74 +++--
.../test_tir_transform_storage_rewrite.py | 4 +-
.../tvmscript/test_tvmscript_ir_builder_tir.py | 12 +-
.../tvmscript/test_tvmscript_printer_annotation.py | 9 +-
.../python/tvmscript/test_tvmscript_printer_tir.py | 15 +-
tests/python/tvmscript/test_tvmscript_roundtrip.py | 318 +-----------------
.../tvmscript/test_tvmscript_syntax_sugar.py | 13 +-
web/src/cache_state.ts | 175 ++++++++++
web/src/index.ts | 1 +
web/src/runtime.ts | 14 +-
web/src/webgpu.ts | 217 +++++++++---
107 files changed, 2142 insertions(+), 1564 deletions(-)
create mode 100644 web/src/cache_state.ts