This is an automated email from the ASF dual-hosted git repository.
github-actions[bot] pushed a change to branch nightly
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 275114b327 [REFACTOR][IR] Unify PrimExpr with Expr typed view (#19910)
add 99869414de [TIRX] Remove SizeVar in favor of contextual constraints
(#19930)
add d27ed727ce [TIRx] Generalize expression functor signatures (#19931)
No new revisions were added by this update.
Summary of changes:
docs/arch/codegen.rst | 2 +-
include/tvm/arith/analyzer.h | 4 +
include/tvm/tirx/expr_functor.h | 86 +++++++-------
include/tvm/tirx/script/builder/ir.h | 37 ++----
include/tvm/tirx/stmt_functor.h | 21 ++--
include/tvm/tirx/var.h | 48 --------
.../tvm/relax/frontend/nn/llm/_decode_kernels.py | 64 +++++-----
python/tvm/relax/frontend/nn/llm/_page_kernels.py | 12 +-
.../tvm/relax/frontend/nn/llm/_prefill_kernels.py | 124 ++++++++++---------
python/tvm/relax/frontend/nn/llm/tree_attn.py | 112 ++++++++---------
python/tvm/relax/frontend/nn/op.py | 6 +-
python/tvm/relax/frontend/onnx/onnx_frontend.py | 52 ++++----
.../frontend/torch/exported_program_translator.py | 12 +-
python/tvm/te/__init__.py | 2 +-
python/tvm/te/operation.py | 22 ----
python/tvm/tirx/__init__.py | 2 +-
python/tvm/tirx/expr.py | 22 ----
python/tvm/tirx/expr_functor.py | 13 --
python/tvm/tirx/functor.py | 38 ------
python/tvm/tirx/script/builder/ir.py | 38 ++----
python/tvm/tirx/stmt_functor.py | 18 ++-
python/tvm/topi/gpu/sort.py | 4 +-
python/tvm/topi/nn/batch_matmul.py | 2 +-
python/tvm/topi/signal.py | 2 +-
python/tvm/topi/sort.py | 2 +-
python/tvm/topi/utils.py | 4 +-
src/arith/bound_deducer.cc | 8 +-
src/arith/canonical_simplify.cc | 75 ++++++------
src/arith/const_int_bound.cc | 24 ++--
src/arith/detect_linear_equation.cc | 2 +-
src/arith/int_set.cc | 6 +-
src/arith/ir_mutator_with_analyzer.cc | 33 +++---
src/arith/ir_mutator_with_analyzer.h | 8 +-
src/arith/iter_affine_map.cc | 38 +++---
src/arith/modular_set.cc | 2 +-
src/arith/rewrite_simplify.cc | 102 ++++++++--------
src/arith/rewrite_simplify.h | 51 ++++----
src/arith/solve_linear_inequality.cc | 23 ++--
src/arith/unwrap_vector_expr.cc | 14 +--
src/arith/z3_prover.cc | 7 +-
src/backend/cuda/codegen/codegen_cuda.cc | 16 +--
src/backend/trn/transform/lower_trainium_layout.cc | 12 +-
src/backend/vulkan/codegen/codegen_spirv.h | 2 +-
src/relax/analysis/layout_transformation.cc | 2 +-
src/relax/analysis/type_analysis.cc | 8 +-
.../transform/lower_global_view_to_local_view.cc | 2 +-
src/relax/script/printer/tir.cc | 1 -
src/relax/transform/attach_global_symbol.cc | 2 +-
src/relax/transform/dataflow_inplace.cc | 2 +-
src/relax/transform/fuse_tir.cc | 18 +--
src/relax/transform/split_call_tir_by_pattern.cc | 3 +-
src/relax/transform/utils.h | 4 +-
src/s_tir/analysis/estimate_flops.cc | 5 +-
src/s_tir/backend/adreno/texture_flatten.cc | 4 +-
.../feature_extractor/per_store_feature.cc | 4 +-
src/s_tir/schedule/ir_comparator.cc | 3 +-
src/s_tir/schedule/ir_comparator.h | 4 +-
src/s_tir/schedule/primitive/block_annotate.cc | 2 +-
src/s_tir/schedule/primitive/blockize_tensorize.cc | 10 +-
.../schedule/primitive/cache_index_helpers.cc | 11 +-
src/s_tir/schedule/primitive/cache_index_helpers.h | 4 +-
src/s_tir/schedule/primitive/cache_read_write.cc | 14 +--
src/s_tir/schedule/primitive/compute_inline.cc | 36 +++---
.../schedule/primitive/layout_transformation.cc | 4 +-
.../schedule/primitive/loop_transformation.cc | 2 +-
src/s_tir/schedule/primitive/pad_einsum.cc | 2 +-
src/s_tir/schedule/primitive/read_write_at.cc | 2 +-
src/s_tir/schedule/primitive/rolling_buffer.cc | 2 +-
src/s_tir/schedule/transform.cc | 10 +-
src/s_tir/schedule/transform.h | 6 +-
src/s_tir/transform/bound_checker.cc | 4 +-
src/s_tir/transform/canonicalize_loop.cc | 2 +-
src/s_tir/transform/compact_buffer_region.cc | 2 +-
src/s_tir/transform/convert_blocks_to_opaque.cc | 8 +-
src/s_tir/transform/hoist_expression.cc | 2 +-
src/s_tir/transform/inject_double_buffer.cc | 4 +-
src/s_tir/transform/inject_permuted_layout.cc | 4 +-
src/s_tir/transform/inject_software_pipeline.cc | 4 +-
src/s_tir/transform/inject_virtual_thread.cc | 22 ++--
src/s_tir/transform/lift_thread_binding.cc | 2 +-
src/s_tir/transform/loop_partition.cc | 10 +-
.../transform/lower_cross_thread_reduction.cc | 4 +-
src/s_tir/transform/lower_match_buffer.cc | 6 +-
src/s_tir/transform/lower_opaque_block.cc | 8 +-
src/s_tir/transform/lower_thread_allreduce.cc | 2 +-
.../transform/memhammer_intermediate_stage.cc | 2 +-
src/s_tir/transform/memhammer_lower_auto_copy.cc | 2 +-
.../transform/merge_shared_memory_allocations.cc | 10 +-
src/s_tir/transform/renew_defs.cc | 27 ++---
src/s_tir/transform/renormalize_split_pattern.cc | 24 ++--
src/s_tir/transform/rewrite_unsafe_select.cc | 6 +-
src/s_tir/transform/transform_mma_buffer_layout.cc | 4 +-
src/s_tir/transform/unify_thread_binding.cc | 2 +-
.../transform/using_assume_to_reduce_branches.cc | 12 +-
src/target/llvm/codegen_llvm.h | 2 +-
src/target/source/codegen_c.h | 2 +-
src/te/operation/compute_op.cc | 2 +-
src/te/operation/create_primfunc.cc | 10 +-
src/tirx/analysis/check_contains.cc | 4 +-
src/tirx/analysis/check_contains.h | 2 +-
src/tirx/analysis/deep_equal.cc | 10 +-
src/tirx/analysis/expr_complexity.cc | 2 +-
src/tirx/analysis/side_effect.cc | 2 +-
src/tirx/analysis/var_touch.cc | 2 +-
src/tirx/analysis/verify_memory.cc | 2 +-
src/tirx/analysis/verify_ssa.cc | 2 +-
src/tirx/analysis/verify_well_formed.cc | 2 +-
src/tirx/ir/data_type_rewriter.cc | 99 ++++++++--------
src/tirx/ir/data_type_rewriter.h | 66 +++++------
src/tirx/ir/expr.cc | 42 +------
src/tirx/ir/expr_functor.cc | 101 +++++++---------
src/tirx/ir/py_functor.cc | 42 +++----
src/tirx/ir/specialize.cc | 30 ++---
src/tirx/ir/stmt_functor.cc | 81 +++++++------
src/tirx/ir/tir_visitor_with_path.cc | 4 -
src/tirx/ir/tir_visitor_with_path.h | 6 +-
src/tirx/script/builder/ir.cc | 12 +-
src/tirx/script/printer/expr.cc | 11 --
src/tirx/transform/bind_target.cc | 2 +-
src/tirx/transform/common_subexpr_elim.cc | 14 ++-
src/tirx/transform/flatten_buffer.cc | 2 +-
src/tirx/transform/force_narrow_index_to_i32.cc | 2 +-
src/tirx/transform/inline_private_functions.cc | 2 +-
src/tirx/transform/ir_utils.cc | 43 +++----
src/tirx/transform/lower_intrin.cc | 38 +++---
src/tirx/transform/lower_tirx_cleanup.cc | 12 +-
src/tirx/transform/lower_tirx_dedup_tensormap.cc | 12 +-
src/tirx/transform/lower_tirx_opaque.cc | 6 +-
src/tirx/transform/lower_tvm_builtin.cc | 22 ++--
src/tirx/transform/lower_warp_memory.cc | 6 +-
src/tirx/transform/make_packed_api.cc | 2 +-
src/tirx/transform/narrow_datatype.cc | 30 +++--
src/tirx/transform/remap_thread_axis.cc | 2 +-
src/tirx/transform/replace_selected_expr.cc | 12 +-
src/tirx/transform/replace_selected_expr.h | 2 +-
src/tirx/transform/split_host_device.cc | 4 +-
src/tirx/transform/stmt_simplify.cc | 13 +-
src/tirx/transform/storage_rewrite.cc | 20 ++--
src/tirx/transform/tvm_ffi_binder.cc | 2 +-
src/tirx/transform/unroll_loop.cc | 2 +-
src/tirx/transform/unsupported_dtype_legalize.cc | 92 +++++++-------
src/tirx/transform/update_pointer_storage_scope.cc | 4 +-
src/tirx/transform/update_pointer_storage_scope.h | 4 +-
src/tirx/transform/vectorize_loop.cc | 132 +++++++++++----------
tests/cpp/ir_functor_test.cc | 8 +-
.../python/arith/test_arith_canonical_simplify.py | 2 +-
tests/python/arith/test_arith_const_int_bound.py | 12 +-
tests/python/arith/test_arith_iter_affine_map.py | 14 +--
tests/python/arith/test_arith_simplify.py | 9 +-
tests/python/codegen/test_target_codegen_arm.py | 4 +-
tests/python/codegen/test_target_codegen_device.py | 2 +-
tests/python/codegen/test_target_codegen_llvm.py | 10 +-
tests/python/codegen/test_target_codegen_vulkan.py | 4 +-
tests/python/ir/test_datatype_nv_fp4.py | 4 +-
tests/python/ir/test_datatype_nv_fp8.py | 4 +-
.../relax/test_frontend_from_exported_program.py | 30 ++---
tests/python/relax/test_frontend_nn_op.py | 6 +-
tests/python/relax/test_frontend_onnx.py | 112 +++++++++--------
tests/python/relax/test_op_image.py | 2 +-
tests/python/relax/test_op_index.py | 32 -----
.../python/relax/test_runtime_builtin_rnn_state.py | 4 +-
..._transform_legalize_ops_index_linear_algebra.py | 8 +-
tests/python/runtime/test_runtime_module_load.py | 2 +-
.../s_tir/dlight/test_gpu_general_reduction.py | 10 +-
tests/python/te/test_te_tag.py | 30 ++---
tests/python/te/test_te_tensor.py | 50 ++++----
tests/python/te/test_te_verify_compute.py | 4 +-
.../test_tir_analysis_undefined_vars.py | 6 +-
.../test_tir_analysis_verify_well_formed.py | 4 +-
tests/python/tirx-base/test_tir_buffer.py | 44 +++----
tests/python/tirx-base/test_tir_expr_functor.py | 14 ---
.../test_tir_transform_convert_ssa.py | 4 +-
.../test_tir_transform_split_host_device.py | 6 +-
.../tirx/transform/test_tirx_expr_functor.py | 14 ---
.../python/tvmscript/test_tvmscript_printer_tir.py | 10 --
tests/python/tvmscript/test_tvmscript_roundtrip.py | 8 +-
176 files changed, 1353 insertions(+), 1652 deletions(-)
