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 2d1f1cd34a [REFACTOR] Update CHECK and ICHECK_GE to TVM_FFI_ICHECK and
TVM_FFI_ICHECK_GE in thrust.cu (#18825)
add e5f483cd5f [REFACTOR][NODE] Remove node redirect headers (#18829)
add 6653620075 [ROCm] Fix some ROCm codegen bugs (#15518)
add c1d32438a0 [BugFix][TOPI] Fix resize accuracy issue with non-floor
rounding (#18838)
add 95a4b1a819 [IR][TIR] Remove body from AssertStmt (#18832)
add 9a8320acbd [LINT][PYTHON] Modernize annotations with ruff UP rules
(#18830)
No new revisions were added by this update.
Summary of changes:
.github/actions/setup/action.yml | 4 +-
ci/jenkins/generate.py | 5 +-
ci/scripts/github/github_cc_reviewers.py | 3 +-
ci/scripts/github/github_commenter.py | 12 +-
ci/scripts/github/github_docs_comment.py | 6 +-
ci/scripts/github/github_skipped_tests_comment.py | 10 +-
ci/scripts/github/github_tag_teams.py | 16 +-
ci/scripts/github/github_tvmbot.py | 37 +--
ci/scripts/github/ping_reviewers.py | 5 +-
ci/scripts/github/update_branch.py | 6 +-
ci/scripts/jenkins/check_pr.py | 5 +-
ci/scripts/jenkins/cmd_utils.py | 3 +-
ci/scripts/jenkins/determine_docker_images.py | 4 +-
ci/scripts/jenkins/git_skip_ci_globs.py | 3 +-
ci/scripts/jenkins/git_utils.py | 22 +-
ci/scripts/jenkins/http_utils.py | 4 +-
ci/scripts/jenkins/open_docker_update_pr.py | 13 +-
ci/scripts/jenkins/pytest_wrapper.py | 9 +-
ci/scripts/jenkins/s3.py | 7 +-
ci/scripts/jenkins/should_rebuild_docker.py | 8 +-
ci/scripts/jenkins/should_run_slow_tests.py | 3 +-
docs/conf.py | 3 +-
docs/download_3rdparty_embeds.py | 34 +--
docs/how_to/tutorials/optimize_llm.py | 5 +-
include/tvm/arith/analyzer.h | 2 +-
include/tvm/ir/attrs.h | 4 +-
include/tvm/ir/env_func.h | 2 +-
include/tvm/ir/expr.h | 4 +-
include/tvm/ir/instrument.h | 2 +-
include/tvm/ir/module.h | 1 +
include/tvm/ir/scope_stack.h | 123 +++++++++
include/tvm/{node => ir}/serialization.h | 24 +-
include/tvm/ir/source_map.h | 2 +-
include/tvm/ir/type.h | 1 -
include/tvm/node/node.h | 63 -----
include/tvm/node/reflection.h | 41 ---
include/tvm/node/serialization.h | 27 +-
include/tvm/node/structural_equal.h | 90 +------
include/tvm/node/structural_hash.h | 108 +-------
include/tvm/relax/distributed/axis_group_graph.h | 4 +-
include/tvm/relax/exec_builder.h | 5 +-
include/tvm/relax/expr.h | 2 +-
include/tvm/relax/struct_info.h | 3 +-
include/tvm/runtime/object.h | 3 +
include/tvm/s_tir/meta_schedule/arg_info.h | 1 -
include/tvm/script/ir_builder/base.h | 2 +-
include/tvm/script/ir_builder/ir/frame.h | 1 -
include/tvm/script/ir_builder/ir/ir.h | 1 -
include/tvm/script/printer/doc.h | 1 -
include/tvm/script/printer/ir_docsifier.h | 2 +-
include/tvm/script/printer/ir_docsifier_functor.h | 2 +-
include/tvm/support/with.h | 74 ++++++
include/tvm/target/tag.h | 1 -
include/tvm/target/target.h | 2 +-
include/tvm/target/target_kind.h | 2 +-
include/tvm/target/virtual_device.h | 2 +-
include/tvm/tir/expr.h | 1 -
include/tvm/tir/function.h | 1 +
include/tvm/tir/index_map.h | 1 +
include/tvm/tir/stmt.h | 3 +-
include/tvm/tir/var.h | 1 -
pyproject.toml | 14 +-
python/tvm/arith/analyzer.py | 5 +-
python/tvm/arith/pattern.py | 4 +-
python/tvm/contrib/cc.py | 3 +-
python/tvm/contrib/cutlass/build.py | 6 +-
python/tvm/contrib/cutlass/library.py | 4 +-
python/tvm/contrib/download.py | 2 +-
python/tvm/contrib/hexagon/build.py | 33 +--
python/tvm/contrib/hexagon/meta_schedule.py | 16 +-
python/tvm/contrib/hexagon/pytest_plugin.py | 5 +-
python/tvm/contrib/hexagon/session.py | 19 +-
python/tvm/contrib/hexagon/tools.py | 9 +-
python/tvm/contrib/ndk.py | 5 +-
python/tvm/contrib/nvcc.py | 7 +-
python/tvm/contrib/rpc.py | 2 -
python/tvm/contrib/tar.py | 2 -
python/tvm/contrib/tvmjs.py | 15 +-
python/tvm/contrib/utils.py | 2 +-
python/tvm/driver/build_module.py | 18 +-
python/tvm/exec/disco_worker.py | 2 +-
python/tvm/ir/analysis.py | 4 +-
python/tvm/ir/expr.py | 12 +-
python/tvm/ir/function.py | 3 +-
python/tvm/ir/global_info.py | 2 +-
python/tvm/ir/instrument.py | 3 +-
python/tvm/ir/module.py | 18 +-
python/tvm/ir/supply.py | 2 +-
python/tvm/ir/transform.py | 12 +-
python/tvm/relax/analysis/analysis.py | 44 ++--
python/tvm/relax/analysis/estimate_memory_usage.py | 3 +-
.../relax/backend/adreno/transform/transform.py | 4 +-
python/tvm/relax/backend/contrib/nnapi.py | 29 +--
python/tvm/relax/backend/cuda/cublas.py | 10 +-
python/tvm/relax/backend/cuda/cudnn.py | 2 +-
python/tvm/relax/backend/cuda/cutlass.py | 4 +-
python/tvm/relax/backend/cuda/flashinfer.py | 13 +-
python/tvm/relax/backend/dispatch_sort_scan.py | 3 +-
python/tvm/relax/backend/gpu_generic/cumsum.py | 3 +-
python/tvm/relax/backend/gpu_generic/sampling.py | 4 +-
python/tvm/relax/backend/pattern_registry.py | 22 +-
python/tvm/relax/backend/patterns.py | 28 +-
python/tvm/relax/backend/rocm/hipblas.py | 4 +-
python/tvm/relax/backend/utils.py | 4 +-
python/tvm/relax/base_py_module.py | 50 ++--
python/tvm/relax/binding_rewrite.py | 4 +-
python/tvm/relax/block_builder.py | 29 ++-
python/tvm/relax/distributed/global_info.py | 10 +-
python/tvm/relax/distributed/struct_info.py | 3 +-
python/tvm/relax/dpl/context.py | 4 +-
python/tvm/relax/dpl/pattern.py | 71 +++--
python/tvm/relax/dpl/rewrite.py | 10 +-
python/tvm/relax/exec_builder.py | 11 +-
python/tvm/relax/expr.py | 191 +++++++-------
python/tvm/relax/expr_functor.py | 114 ++++----
python/tvm/relax/frontend/common.py | 4 +-
python/tvm/relax/frontend/nn/_tensor_op.py | 8 +-
python/tvm/relax/frontend/nn/core.py | 93 +++----
python/tvm/relax/frontend/nn/exporter.py | 40 +--
python/tvm/relax/frontend/nn/extern.py | 24 +-
python/tvm/relax/frontend/nn/llm/kv_cache.py | 50 ++--
.../relax/frontend/nn/llm/position_embedding.py | 33 +--
python/tvm/relax/frontend/nn/llm/tree_attn.py | 14 +-
python/tvm/relax/frontend/nn/modules.py | 86 +++---
python/tvm/relax/frontend/nn/op.py | 189 +++++++-------
python/tvm/relax/frontend/nn/spec.py | 44 ++--
python/tvm/relax/frontend/nn/subroutine.py | 2 +-
python/tvm/relax/frontend/nn/torch.py | 13 +-
python/tvm/relax/frontend/onnx/onnx_frontend.py | 69 ++---
.../frontend/stablehlo/stablehlo_translator.py | 19 +-
.../frontend/torch/base_fx_graph_translator.py | 204 +++++++--------
python/tvm/relax/frontend/torch/dynamo.py | 3 +-
.../frontend/torch/exported_program_translator.py | 38 ++-
python/tvm/relax/frontend/torch/fx_translator.py | 18 +-
python/tvm/relax/op/_op_gradient.py | 101 ++++----
python/tvm/relax/op/base.py | 66 ++---
python/tvm/relax/op/builtin/builtin.py | 8 +-
python/tvm/relax/op/create.py | 78 +++---
python/tvm/relax/op/datatype.py | 6 +-
python/tvm/relax/op/distributed/distributed.py | 8 +-
python/tvm/relax/op/grad/grad.py | 26 +-
python/tvm/relax/op/image/image.py | 18 +-
python/tvm/relax/op/index.py | 8 +-
python/tvm/relax/op/linear_algebra.py | 10 +-
python/tvm/relax/op/manipulate.py | 62 ++---
python/tvm/relax/op/memory/memory.py | 14 +-
python/tvm/relax/op/memory/view.py | 10 +-
python/tvm/relax/op/nn/nn.py | 170 ++++++------
python/tvm/relax/op/search.py | 6 +-
python/tvm/relax/op/set.py | 14 +-
python/tvm/relax/op/statistical.py | 26 +-
python/tvm/relax/op/vm/vm.py | 16 +-
python/tvm/relax/pipeline.py | 5 +-
python/tvm/relax/relax_to_pyfunc_converter.py | 98 ++++---
python/tvm/relax/struct_info.py | 40 +--
python/tvm/relax/testing/ast_printer.py | 2 +-
python/tvm/relax/testing/lib_comparator.py | 6 +-
python/tvm/relax/testing/nn.py | 38 ++-
python/tvm/relax/testing/transform.py | 15 +-
python/tvm/relax/testing/vm.py | 4 +-
python/tvm/relax/training/loss.py | 26 +-
python/tvm/relax/training/optimizer.py | 18 +-
python/tvm/relax/training/setup_trainer.py | 8 +-
python/tvm/relax/training/trainer.py | 34 ++-
python/tvm/relax/training/utils.py | 6 +-
.../tvm/relax/transform/attach_external_modules.py | 4 +-
.../tvm/relax/transform/ipc_allreduce_rewrite.py | 10 +-
.../tvm/relax/transform/lazy_transform_params.py | 8 +-
python/tvm/relax/transform/legalize_ops/common.py | 6 +-
python/tvm/relax/transform/legalize_ops/create.py | 6 +-
.../relax/transform/legalize_ops/linear_algebra.py | 4 +-
.../tvm/relax/transform/legalize_ops/manipulate.py | 16 +-
python/tvm/relax/transform/legalize_ops/nn.py | 3 +-
python/tvm/relax/transform/legalize_ops/qdq.py | 12 +-
.../relax/transform/legalize_ops/statistical.py | 14 +-
python/tvm/relax/transform/transform.py | 85 +++---
python/tvm/relax/type_converter.py | 11 +-
python/tvm/relax/utils.py | 34 +--
python/tvm/relax/vm_build.py | 30 +--
python/tvm/rpc/client.py | 4 +-
python/tvm/rpc/server_ios_launcher.py | 16 +-
python/tvm/runtime/_tensor.py | 8 +-
python/tvm/runtime/disco/session.py | 19 +-
python/tvm/runtime/executable.py | 9 +-
python/tvm/runtime/module.py | 4 +-
python/tvm/runtime/params.py | 2 +-
python/tvm/runtime/profiling/__init__.py | 9 +-
python/tvm/runtime/script_printer.py | 56 ++--
python/tvm/runtime/support.py | 2 +-
python/tvm/runtime/vm.py | 35 +--
python/tvm/s_tir/analysis/__init__.py | 26 +-
python/tvm/s_tir/data_layout.py | 6 +-
python/tvm/s_tir/dlight/adreno/convolution.py | 8 +-
python/tvm/s_tir/dlight/adreno/fallback.py | 8 +-
python/tvm/s_tir/dlight/adreno/layout_transform.py | 8 +-
python/tvm/s_tir/dlight/adreno/utils.py | 5 +-
.../tvm/s_tir/dlight/analysis/common_analysis.py | 38 ++-
python/tvm/s_tir/dlight/analysis/gemv.py | 8 +-
python/tvm/s_tir/dlight/base/common_schedules.py | 10 +-
python/tvm/s_tir/dlight/base/schedule_rule.py | 8 +-
python/tvm/s_tir/dlight/base/transform.py | 6 +-
python/tvm/s_tir/dlight/base/utils.py | 12 +-
python/tvm/s_tir/dlight/benchmark/bench.py | 49 ++--
python/tvm/s_tir/dlight/benchmark/extract.py | 39 ++-
python/tvm/s_tir/dlight/benchmark/utils.py | 14 +-
python/tvm/s_tir/dlight/cpu/gemv.py | 8 +-
python/tvm/s_tir/dlight/gpu/fallback.py | 10 +-
python/tvm/s_tir/dlight/gpu/gemv.py | 15 +-
python/tvm/s_tir/dlight/gpu/general_reduction.py | 4 +-
python/tvm/s_tir/dlight/gpu/low_batch_gemv.py | 28 +-
python/tvm/s_tir/dlight/gpu/matmul.py | 35 ++-
python/tvm/s_tir/dlight/gpu/reduction.py | 16 +-
python/tvm/s_tir/dlight/gpu/rmsnorm.py | 4 +-
python/tvm/s_tir/dlight/gpu/transpose.py | 4 +-
python/tvm/s_tir/meta_schedule/arg_info.py | 8 +-
python/tvm/s_tir/meta_schedule/builder/builder.py | 23 +-
.../s_tir/meta_schedule/builder/local_builder.py | 35 +--
.../s_tir/meta_schedule/cost_model/cost_model.py | 29 ++-
.../s_tir/meta_schedule/cost_model/mlp_model.py | 50 ++--
.../s_tir/meta_schedule/cost_model/random_model.py | 18 +-
.../s_tir/meta_schedule/cost_model/xgb_model.py | 64 ++---
.../tvm/s_tir/meta_schedule/database/database.py | 81 +++---
.../s_tir/meta_schedule/database/json_database.py | 7 +-
.../meta_schedule/database/schedule_fn_database.py | 2 +-
python/tvm/s_tir/meta_schedule/extracted_task.py | 6 +-
.../feature_extractor/feature_extractor.py | 15 +-
.../feature_extractor/random_feature_extractor.py | 8 +-
python/tvm/s_tir/meta_schedule/logging.py | 25 +-
.../measure_callback/measure_callback.py | 23 +-
python/tvm/s_tir/meta_schedule/mutator/mutator.py | 19 +-
.../s_tir/meta_schedule/post_optimization/space.py | 10 +-
.../s_tir/meta_schedule/post_optimization/utils.py | 5 +-
.../tvm/s_tir/meta_schedule/postproc/postproc.py | 15 +-
python/tvm/s_tir/meta_schedule/profiler.py | 4 +-
.../tvm/s_tir/meta_schedule/relax_integration.py | 44 ++--
python/tvm/s_tir/meta_schedule/runner/config.py | 10 +-
.../tvm/s_tir/meta_schedule/runner/local_runner.py | 56 ++--
.../tvm/s_tir/meta_schedule/runner/rpc_runner.py | 76 +++---
python/tvm/s_tir/meta_schedule/runner/runner.py | 25 +-
python/tvm/s_tir/meta_schedule/runner/utils.py | 21 +-
.../schedule/cuda/layout_transform.py | 39 ++-
.../meta_schedule/schedule_rule/add_rfactor.py | 4 +-
.../s_tir/meta_schedule/schedule_rule/auto_bind.py | 4 +-
.../meta_schedule/schedule_rule/auto_inline.py | 4 +-
.../schedule_rule/cross_thread_reduction.py | 4 +-
.../schedule_rule/multi_level_tiling.py | 47 ++--
.../schedule_rule/parallel_vectorize_unroll.py | 4 +-
.../meta_schedule/schedule_rule/schedule_rule.py | 19 +-
.../search_strategy/search_strategy.py | 37 +--
.../space_generator/space_generator.py | 65 +++--
.../space_generator/space_generator_union.py | 4 +-
.../meta_schedule/task_scheduler/task_scheduler.py | 43 +--
.../meta_schedule/testing/custom_builder_runner.py | 6 +-
.../s_tir/meta_schedule/testing/dummy_object.py | 7 +-
.../meta_schedule/testing/space_generation.py | 27 +-
.../tvm/s_tir/meta_schedule/testing/te_workload.py | 35 ++-
python/tvm/s_tir/meta_schedule/testing/tune_te.py | 3 +-
.../tvm/s_tir/meta_schedule/testing/tune_utils.py | 12 +-
.../meta_schedule/testing/validate_database.py | 57 ++--
python/tvm/s_tir/meta_schedule/tir_integration.py | 34 +--
python/tvm/s_tir/meta_schedule/tune.py | 10 +-
python/tvm/s_tir/meta_schedule/tune_context.py | 30 +--
python/tvm/s_tir/meta_schedule/utils.py | 17 +-
python/tvm/s_tir/sblock_dependence_info.py | 6 +-
python/tvm/s_tir/sblock_scope.py | 8 +-
python/tvm/s_tir/schedule/_type_checker.py | 90 ++++---
python/tvm/s_tir/schedule/analysis.py | 12 +-
python/tvm/s_tir/schedule/instruction.py | 18 +-
python/tvm/s_tir/schedule/schedule.py | 287 ++++++++++-----------
python/tvm/s_tir/schedule/state.py | 15 +-
python/tvm/s_tir/schedule/testing.py | 9 +-
python/tvm/s_tir/schedule/trace.py | 32 +--
python/tvm/s_tir/schedule/transform.py | 4 +-
python/tvm/s_tir/tensor_intrin/cuda.py | 24 +-
python/tvm/s_tir/tensor_intrin/metal.py | 12 +-
python/tvm/script/highlight.py | 14 +-
python/tvm/script/ir_builder/base.py | 9 +-
python/tvm/script/ir_builder/ir/ir.py | 14 +-
.../tvm/script/ir_builder/relax/distributed/ir.py | 20 +-
python/tvm/script/ir_builder/relax/ir.py | 27 +-
.../tvm/script/ir_builder/tir/external_kernel.py | 22 +-
python/tvm/script/ir_builder/tir/frame.py | 4 +-
python/tvm/script/ir_builder/tir/ir.py | 157 +++++------
python/tvm/script/ir_builder/tir/triton.py | 12 +-
python/tvm/script/ir_builder/tir/utils.py | 5 +-
python/tvm/script/parser/core/diagnostics.py | 3 +-
python/tvm/script/parser/core/dispatch.py | 21 +-
python/tvm/script/parser/core/doc.py | 23 +-
python/tvm/script/parser/core/entry.py | 12 +-
python/tvm/script/parser/core/evaluator.py | 56 ++--
python/tvm/script/parser/core/parser.py | 43 +--
python/tvm/script/parser/core/utils.py | 11 +-
python/tvm/script/parser/ir/entry.py | 4 +-
python/tvm/script/parser/relax/dist.py | 16 +-
python/tvm/script/parser/relax/entry.py | 112 ++++----
python/tvm/script/parser/relax/parser.py | 16 +-
python/tvm/script/parser/tir/entry.py | 22 +-
python/tvm/script/parser/tir/operation.py | 6 +-
python/tvm/script/parser/tir/parser.py | 20 +-
python/tvm/script/printer/doc.py | 73 +++---
python/tvm/script/printer/doc_printer.py | 6 +-
python/tvm/target/codegen.py | 4 +-
python/tvm/target/detect_target.py | 4 +-
python/tvm/target/tag_registry/registry.py | 6 +-
python/tvm/target/target.py | 12 +-
python/tvm/te/operation.py | 13 +-
python/tvm/testing/plugin.py | 2 +-
python/tvm/testing/runner.py | 23 +-
python/tvm/testing/utils.py | 18 +-
python/tvm/tir/analysis/analysis.py | 5 +-
python/tvm/tir/buffer.py | 8 +-
python/tvm/tir/build.py | 14 +-
python/tvm/tir/expr.py | 130 +++++-----
python/tvm/tir/function.py | 27 +-
python/tvm/tir/functor.py | 194 +++++++-------
python/tvm/tir/op.py | 16 +-
python/tvm/tir/pipeline.py | 4 +-
python/tvm/tir/stmt.py | 131 +++++-----
python/tvm/tir/transform/function_pass.py | 12 +-
python/tvm/tir/transform/transform.py | 2 +-
python/tvm/topi/broadcast.py | 2 -
python/tvm/topi/gpu/scan.py | 28 +-
python/tvm/topi/image/__init__.py | 2 -
python/tvm/topi/image/dilation2d.py | 2 -
python/tvm/topi/image/resize.py | 5 +-
python/tvm/topi/index_put.py | 2 +-
python/tvm/topi/nn/__init__.py | 2 -
python/tvm/topi/nn/batch_norm.py | 15 +-
python/tvm/topi/nn/bitserial_conv2d.py | 4 +-
python/tvm/topi/nn/bnn.py | 2 -
python/tvm/topi/nn/conv1d_transpose.py | 4 +-
python/tvm/topi/nn/conv2d.py | 30 +--
python/tvm/topi/nn/conv3d_transpose.py | 2 +-
python/tvm/topi/nn/correlation.py | 2 +-
python/tvm/topi/nn/depthwise_conv2d.py | 6 +-
python/tvm/topi/nn/elemwise.py | 2 -
python/tvm/topi/nn/fifo_buffer.py | 2 -
python/tvm/topi/nn/mapping.py | 2 -
python/tvm/topi/nn/pad.py | 2 -
python/tvm/topi/nn/utils.py | 8 +-
python/tvm/topi/reduction.py | 4 +-
python/tvm/topi/scan.py | 20 +-
python/tvm/topi/tensor.py | 7 +-
python/tvm/topi/testing/__init__.py | 2 -
python/tvm/topi/testing/attention_python.py | 6 +-
python/tvm/topi/testing/conv1d_ncw_python.py | 4 +-
python/tvm/topi/testing/matrix_set_diag.py | 2 +-
python/tvm/topi/testing/poolnd_python.py | 27 +-
python/tvm/topi/testing/resize_python.py | 55 +++-
python/tvm/topi/transform.py | 4 +-
python/tvm/topi/utils.py | 4 +-
python/tvm/topi/vision/nms.py | 2 +-
src/arith/conjunctive_normal_form.cc | 4 +-
src/arith/ir_mutator_with_analyzer.cc | 146 ++++++-----
src/arith/ir_mutator_with_analyzer.h | 5 +
src/arith/ir_visitor_with_analyzer.cc | 79 +++---
src/arith/ir_visitor_with_analyzer.h | 6 +
src/arith/iter_affine_map.cc | 6 +-
src/arith/solve_linear_inequality.cc | 4 +-
src/arith/transitive_comparison_analyzer.cc | 2 +-
src/contrib/msc/core/printer/msc_base_printer.cc | 1 +
src/ir/module.cc | 2 +-
src/{node => ir}/serialization.cc | 2 +-
src/{node => ir}/structural_equal.cc | 10 +-
src/{node => ir}/structural_hash.cc | 8 +-
src/ir/transform.cc | 6 +-
src/node/reflection.cc | 1 -
src/node/script_printer.cc | 1 +
src/relax/analysis/struct_info_analysis.cc | 8 +-
.../backend/adreno/annotate_custom_storage.cc | 2 +-
.../backend/adreno/fold_vdevice_scope_change.cc | 2 +-
src/relax/backend/vm/vm_shape_lower.cc | 2 +-
.../distributed/transform/legalize_redistribute.cc | 2 +-
src/relax/distributed/transform/lower_distir.cc | 3 +-
.../distributed/transform/propagate_sharding.cc | 9 +-
src/relax/ir/block_builder.cc | 4 +-
src/relax/ir/dataflow_block_rewriter.cc | 2 +-
src/relax/ir/dataflow_expr_rewriter.cc | 8 +-
src/relax/ir/dataflow_matcher.cc | 15 +-
src/relax/ir/expr_functor.cc | 2 +-
src/relax/op/distributed/utils.cc | 4 +-
src/relax/op/tensor/inspect.cc | 11 +-
src/relax/op/tensor/manipulate.cc | 2 +-
src/relax/training/utils.cc | 2 +-
src/relax/transform/alter_op_impl.cc | 2 +-
src/relax/transform/canonicalize_bindings.cc | 5 +-
src/relax/transform/convert_layout.cc | 2 +-
src/relax/transform/eliminate_common_subexpr.cc | 6 +-
src/relax/transform/fold_constant.cc | 3 +-
src/relax/transform/fuse_tir.cc | 2 +-
src/relax/transform/kill_after_last_use.cc | 2 +-
src/relax/transform/lift_transform_params.cc | 4 +-
src/relax/transform/remove_unused_outputs.cc | 1 +
.../specialize_primfunc_based_on_callsite.cc | 2 +-
src/s_tir/analysis/estimate_flops.cc | 1 -
src/s_tir/meta_schedule/module_equality.cc | 6 +-
src/s_tir/meta_schedule/utils.h | 7 +-
src/s_tir/schedule/primitive/compute_inline.cc | 2 +-
.../schedule/primitive/layout_transformation.cc | 3 +-
src/s_tir/schedule/utils.h | 2 +-
src/s_tir/transform/bound_checker.cc | 3 +-
src/s_tir/transform/inject_software_pipeline.cc | 3 +-
.../transform/using_assume_to_reduce_branches.cc | 7 +-
src/script/ir_builder/tir/frame.cc | 11 +-
src/script/printer/relax/expr.cc | 1 +
src/script/printer/relax/utils.h | 2 +-
src/script/printer/tir/stmt.cc | 22 +-
src/script/printer/utils.h | 2 +-
src/support/ffi_testing.cc | 1 +
src/support/scalars.h | 1 +
src/target/llvm/codegen_llvm.cc | 7 +-
src/target/source/codegen_c.cc | 1 -
src/target/source/codegen_c_host.cc | 1 -
src/target/source/codegen_metal.cc | 1 +
src/target/source/codegen_webgpu.cc | 3 +-
src/target/spirv/codegen_spirv.cc | 3 +-
src/te/operation/compute_op.cc | 2 +-
src/te/operation/create_primfunc.cc | 2 +-
src/tir/ir/stmt.cc | 10 +-
src/tir/ir/stmt_functor.cc | 5 +-
src/tir/ir/tir_visitor_with_path.cc | 1 -
src/tir/transform/arg_binder.cc | 16 +-
src/tir/transform/common_subexpr_elim_tools.cc | 2 +-
src/tir/transform/common_subexpr_elim_tools.h | 12 +-
src/tir/transform/ir_utils.cc | 7 +-
src/tir/transform/make_packed_api.cc | 12 +-
src/tir/transform/skip_assert.cc | 5 +-
src/tir/transform/split_host_device.cc | 2 +-
src/tir/transform/vectorize_loop.cc | 3 +-
tests/conda/build-environment.yaml | 2 +-
tests/cpp/arith_simplify_test.cc | 5 +-
tests/cpp/expr_test.cc | 2 +-
tests/cpp/nested_msg_test.cc | 12 +-
tests/cpp/target/virtual_device_test.cc | 10 +-
tests/python/arith/test_arith_intset.py | 2 +-
tests/python/ci/test_tvmbot.py | 18 +-
tests/python/ci/test_utils.py | 4 +-
.../python/codegen/test_target_codegen_cuda_fp8.py | 9 +-
tests/python/contrib/test_cutlass_gemm.py | 9 +-
.../python/contrib/test_hexagon/benchmark_util.py | 2 +-
tests/python/contrib/test_hexagon/pytest_util.py | 13 +-
.../test_hexagon/test_2d_physical_buffers.py | 2 +-
.../test_hexagon/test_benchmark_maxpool2d.py | 3 +-
.../contrib/test_msc/test_translate_relax.py | 6 +-
.../contrib/test_msc/test_translate_tensorrt.py | 2 +-
.../contrib/test_msc/test_translate_torch.py | 4 +-
tests/python/disco/test_nvshmem.py | 3 +-
tests/python/ir/analysis/test_collect_call_map.py | 3 +-
tests/python/nightly/test_nnapi/infrastructure.py | 8 +-
tests/python/nightly/test_nnapi/test_network.py | 4 +-
tests/python/nightly/test_nnapi/test_ops.py | 4 +-
.../test_distributed_tvmscript_parser.py | 4 +-
.../test_runtime_builtin_kv_cache_transfer.py | 20 +-
tests/python/relax/test_analysis.py | 3 +-
.../test_analysis_computable_at_compile_time.py | 3 +-
.../python/relax/test_analysis_detect_recursion.py | 3 +-
tests/python/relax/test_ast_printer.py | 3 +-
tests/python/relax/test_dataflow_inplace.py | 3 +-
tests/python/relax/test_expr_args_converter.py | 13 +-
tests/python/relax/test_frontend_nn_jit.py | 7 +-
tests/python/relax/test_frontend_nn_modules.py | 5 +-
tests/python/relax/test_frontend_onnx.py | 42 +--
tests/python/relax/test_frontend_stablehlo.py | 17 +-
tests/python/relax/test_group_gemm_flashinfer.py | 2 +-
tests/python/relax/test_op_binary.py | 2 +-
tests/python/relax/test_op_gradient_numeric.py | 9 +-
tests/python/relax/test_op_search.py | 2 +-
tests/python/relax/test_op_statistical.py | 2 +-
tests/python/relax/test_op_unary.py | 2 +-
...runtime_builtin_paged_attention_kv_cache_cpu.py | 19 +-
..._builtin_paged_attention_kv_cache_flashinfer.py | 7 +-
...ltin_paged_attention_kv_cache_mla_flashinfer.py | 5 +-
...ime_builtin_paged_attention_kv_cache_mla_tir.py | 5 +-
...runtime_builtin_paged_attention_kv_cache_tir.py | 19 +-
.../python/relax/test_runtime_builtin_rnn_state.py | 4 +-
.../relax/test_runtime_sampling_flashinfer.py | 3 +-
.../relax/test_training_optimizer_numeric.py | 12 +-
.../python/relax/test_transform_few_shot_tuning.py | 7 +-
.../relax/test_transform_legalize_ops_image.py | 4 +-
.../relax/test_transform_legalize_ops_unary.py | 2 +-
.../test_transform_update_param_struct_info.py | 5 +-
tests/python/relax/test_tvmscript_parser.py | 4 +-
.../relax/test_tvmscript_parser_op_arith_cmp.py | 6 +-
.../relax/test_tvmscript_parser_op_create.py | 5 +-
.../relax/test_tvmscript_parser_op_datatype.py | 5 +-
.../python/relax/test_tvmscript_parser_op_grad.py | 5 +-
.../python/relax/test_tvmscript_parser_op_image.py | 5 +-
.../python/relax/test_tvmscript_parser_op_index.py | 5 +-
.../test_tvmscript_parser_op_linear_algebra.py | 5 +-
.../relax/test_tvmscript_parser_op_manipulate.py | 5 +-
tests/python/relax/test_tvmscript_parser_op_nn.py | 5 +-
.../relax/test_tvmscript_parser_op_search.py | 5 +-
tests/python/relax/test_tvmscript_parser_op_set.py | 5 +-
.../python/relax/test_tvmscript_parser_op_sort.py | 5 +-
.../relax/test_tvmscript_parser_op_statistical.py | 5 +-
.../relax/test_tvmscript_parser_op_vision.py | 5 +-
tests/python/relax/test_vm_build.py | 4 +-
tests/python/relax/test_vm_multi_device.py | 4 +-
.../meta_schedule/test_meta_schedule_builder.py | 7 +-
.../meta_schedule/test_meta_schedule_cost_model.py | 13 +-
.../meta_schedule/test_meta_schedule_database.py | 31 +--
.../test_meta_schedule_feature_extractor.py | 9 +-
...schedule_feature_extractor_per_store_feature.py | 4 +-
.../test_meta_schedule_measure_callback.py | 23 +-
.../test_meta_schedule_mutator_mutate_parallel.py | 3 +-
.../test_meta_schedule_mutator_mutate_tile_size.py | 3 +-
.../test_meta_schedule_mutator_mutate_unroll.py | 3 +-
.../test_meta_schedule_post_order_apply.py | 11 +-
.../meta_schedule/test_meta_schedule_runner.py | 36 +--
...eta_schedule_schedule_rule_apply_custom_rule.py | 3 +-
.../test_meta_schedule_search_strategy.py | 13 +-
...ule_space_cuda_async_multiple_initialization.py | 10 +-
.../test_meta_schedule_task_scheduler.py | 3 +-
.../s_tir/schedule/test_tir_schedule_analysis.py | 5 +-
.../python/testing/test_type_annotation_checker.py | 25 +-
tests/python/tir-base/test_tir_constructor.py | 3 +-
tests/python/tir-base/test_tir_imm_values.py | 8 +-
tests/python/tir-base/test_tir_ops.py | 2 +-
.../tvmscript/test_tvmscript_ir_builder_tir.py | 9 +-
.../python/tvmscript/test_tvmscript_printer_tir.py | 4 +-
.../test_tvmscript_printer_underlining.py | 3 +-
tests/scripts/ci.py | 41 +--
tests/scripts/release/gather_prs.py | 5 +-
tests/scripts/release/make_notes.py | 1 -
524 files changed, 4325 insertions(+), 4691 deletions(-)
create mode 100644 include/tvm/ir/scope_stack.h
copy include/tvm/{node => ir}/serialization.h (67%)
delete mode 100644 include/tvm/node/node.h
delete mode 100644 include/tvm/node/reflection.h
rename src/{node => ir}/serialization.cc (97%)
rename src/{node => ir}/structural_equal.cc (90%)
rename src/{node => ir}/structural_hash.cc (96%)
