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 c0828bc8ad [REFACTOR][TEST] Migrate tir-transform tests from TE to
TVMScript (#18805)
add aa2e609136 [LINT] Modernize lint to use pre-commit hooks (#18807)
add 7ac12ebdd8 [CI] Add GitHub Actions lint workflow (#18809)
add 55027b7a39 [DRAFT][DO NOT MERGE] Bump tvm-ffi to v0.1.9 (#18801)
No new revisions were added by this update.
Summary of changes:
.asf.yaml | 1 -
.clang-format | 2 +-
.../{update_nightly_branch.yml => lint.yml} | 34 +-
.gitignore | 4 +
.../msc/__init__.py => .markdownlint-cli2.yaml | 4 +-
.pre-commit-config.yaml | 152 +--
...ubuntu_install_cuda_python.sh => .yamllint.yaml | 19 +-
3rdparty/tvm-ffi | 2 +-
Makefile | 16 +-
.../java/org/apache/tvm/tvmrpc/MainActivity.java | 55 +-
.../java/org/apache/tvm/tvmrpc/RPCActivity.java | 24 +-
.../org/apache/tvm/tvmrpc/RPCAndroidWatchdog.java | 3 +-
.../java/org/apache/tvm/tvmrpc/RPCProcessor.java | 3 +-
apps/android_rpc/tests/android_rpc_test.py | 9 +-
apps/ios_rpc/init_proj.py | 6 +-
apps/ios_rpc/tests/ios_rpc_test.py | 1 +
apps/ios_rpc/tvmrpc/AppDelegate.m | 46 +-
.../AppIcon.appiconset/Contents.json | 110 +-
apps/ios_rpc/tvmrpc/RPCServer.mm | 9 +-
apps/ios_rpc/tvmrpc/main.m | 10 +-
ci/jenkins/data.py | 1 -
ci/jenkins/generate.py | 10 +-
ci/scripts/github/__init__.py | 1 +
ci/scripts/github/github_cc_reviewers.py | 10 +-
ci/scripts/github/github_commenter.py | 6 +-
ci/scripts/github/github_docs_comment.py | 4 +-
ci/scripts/github/github_pr_comment.py | 6 +-
ci/scripts/github/github_skipped_tests_comment.py | 28 +-
ci/scripts/github/github_tag_teams.py | 13 +-
ci/scripts/github/github_tvmbot.py | 17 +-
ci/scripts/github/ping_reviewers.py | 4 +-
ci/scripts/github/update_branch.py | 7 +-
ci/scripts/jenkins/check_pr.py | 14 +-
ci/scripts/jenkins/cmd_utils.py | 7 +-
ci/scripts/jenkins/determine_docker_images.py | 7 +-
ci/scripts/jenkins/git_skip_ci.py | 9 +-
ci/scripts/jenkins/git_skip_ci_globs.py | 1 -
ci/scripts/jenkins/git_utils.py | 12 +-
ci/scripts/jenkins/http_utils.py | 2 +-
ci/scripts/jenkins/open_docker_update_pr.py | 8 +-
ci/scripts/jenkins/pytest_ids.py | 6 +-
ci/scripts/jenkins/pytest_wrapper.py | 11 +-
ci/scripts/jenkins/s3.py | 4 +-
ci/scripts/jenkins/should_rebuild_docker.py | 7 +-
ci/scripts/jenkins/should_run_slow_tests.py | 8 +-
conftest.py | 6 +-
docker/lint.sh | 53 +-
docs/.rstcheck.cfg | 5 +
docs/conf.py | 28 +-
docs/deep_dive/tensor_ir/tutorials/tir_creation.py | 2 +-
docs/download_3rdparty_embeds.py | 4 +-
docs/get_started/tutorials/ir_module.py | 10 +-
docs/get_started/tutorials/quick_start.py | 2 +-
docs/how_to/tutorials/cross_compilation_and_rpc.py | 6 +-
docs/how_to/tutorials/customize_opt.py | 4 +-
docs/how_to/tutorials/e2e_opt_model.py | 1 +
docs/how_to/tutorials/optimize_llm.py | 3 +-
include/tvm/ir/type_functor.h | 6 +-
include/tvm/relax/dataflow_pattern_functor.h | 6 +-
include/tvm/relax/expr_functor.h | 6 +-
include/tvm/relax/struct_info_functor.h | 6 +-
include/tvm/tir/expr_functor.h | 6 +-
include/tvm/tir/stmt_functor.h | 6 +-
include/tvm/topi/elemwise.h | 21 +-
jvm/core/src/main/java/org/apache/tvm/API.java | 7 +-
.../src/main/java/org/apache/tvm/APIInternal.java | 3 +-
jvm/core/src/main/java/org/apache/tvm/Base.java | 10 +-
jvm/core/src/main/java/org/apache/tvm/Device.java | 27 +-
.../src/main/java/org/apache/tvm/Function.java | 3 +-
jvm/core/src/main/java/org/apache/tvm/LibInfo.java | 5 +-
jvm/core/src/main/java/org/apache/tvm/Module.java | 16 +-
.../java/org/apache/tvm/NativeLibraryLoader.java | 6 +-
.../src/main/java/org/apache/tvm/TVMObject.java | 3 +-
jvm/core/src/main/java/org/apache/tvm/TVMType.java | 15 +-
.../src/main/java/org/apache/tvm/TVMValue.java | 7 +-
.../main/java/org/apache/tvm/TVMValueBytes.java | 3 +-
.../main/java/org/apache/tvm/TVMValueDouble.java | 3 +-
.../main/java/org/apache/tvm/TVMValueHandle.java | 3 +-
.../src/main/java/org/apache/tvm/TVMValueLong.java | 3 +-
.../src/main/java/org/apache/tvm/TVMValueNull.java | 3 +-
.../main/java/org/apache/tvm/TVMValueString.java | 3 +-
jvm/core/src/main/java/org/apache/tvm/Tensor.java | 3 +-
.../src/main/java/org/apache/tvm/TensorBase.java | 6 +-
.../tvm/rpc/ConnectProxyServerProcessor.java | 9 +-
.../tvm/rpc/ConnectTrackerServerProcessor.java | 32 +-
.../java/org/apache/tvm/rpc/NativeServerLoop.java | 14 +-
jvm/core/src/main/java/org/apache/tvm/rpc/RPC.java | 17 +-
.../main/java/org/apache/tvm/rpc/RPCSession.java | 7 +-
.../main/java/org/apache/tvm/rpc/RPCWatchdog.java | 3 +-
.../src/main/java/org/apache/tvm/rpc/Server.java | 6 +-
.../java/org/apache/tvm/rpc/SocketChannel.java | 11 +-
.../apache/tvm/rpc/StandaloneServerProcessor.java | 6 +-
.../src/test/java/org/apache/tvm/FunctionTest.java | 30 +-
.../src/test/java/org/apache/tvm/ModuleTest.java | 19 +-
.../src/test/java/org/apache/tvm/TensorTest.java | 46 +-
.../src/test/java/org/apache/tvm/TestUtils.java | 4 +-
.../src/test/java/org/apache/tvm/rpc/RPCTest.java | 26 +-
jvm/core/src/test/scripts/prepare_test_libs.py | 6 +-
jvm/core/src/test/scripts/test_rpc_proxy_server.py | 1 +
jvm/pom.xml | 17 +
mypy.ini | 33 -
pyproject.toml | 338 +++---
python/tvm/__init__.py | 6 +-
python/tvm/arith/__init__.py | 1 +
python/tvm/arith/_ffi_api.py | 2 +-
python/tvm/arith/analyzer.py | 4 +-
python/tvm/arith/bound.py | 1 +
python/tvm/arith/int_set.py | 3 +
python/tvm/arith/int_solver.py | 3 +
python/tvm/arith/iter_affine_map.py | 6 +-
python/tvm/arith/pattern.py | 1 +
python/tvm/base.py | 1 +
python/tvm/contrib/cblas.py | 1 +
python/tvm/contrib/cc.py | 1 +
python/tvm/contrib/clang.py | 4 +-
python/tvm/contrib/coreml_runtime.py | 4 +-
python/tvm/contrib/cublas.py | 1 +
python/tvm/contrib/cublaslt.py | 1 +
python/tvm/contrib/cudnn.py | 32 +-
python/tvm/contrib/cutlass/__init__.py | 2 +
python/tvm/contrib/cutlass/_ffi_api.py | 1 +
python/tvm/contrib/cutlass/attention_operation.py | 1 +
python/tvm/contrib/cutlass/build.py | 4 +-
python/tvm/contrib/cutlass/conv2d_operation.py | 7 +-
python/tvm/contrib/cutlass/conv2d_profiler.py | 2 +-
python/tvm/contrib/cutlass/gemm_operation.py | 1 +
python/tvm/contrib/cutlass/gemm_profiler.py | 2 +-
python/tvm/contrib/cutlass/gen_conv2d.py | 14 +-
python/tvm/contrib/cutlass/gen_gemm.py | 1 +
python/tvm/contrib/cutlass/gen_tensor_op.py | 30 +-
python/tvm/contrib/cutlass/layer_norm_operation.py | 1 +
python/tvm/contrib/cutlass/library.py | 5 +-
python/tvm/contrib/cutlass/rms_norm_operation.py | 1 +
python/tvm/contrib/dlpack.py | 1 +
python/tvm/contrib/dnnl.py | 2 +
python/tvm/contrib/download.py | 2 +-
python/tvm/contrib/emcc.py | 1 +
python/tvm/contrib/hexagon/build.py | 16 +-
python/tvm/contrib/hexagon/hexagon_profiler.py | 5 +-
python/tvm/contrib/hexagon/hexagon_unary_ops.py | 2 +
python/tvm/contrib/hexagon/meta_schedule.py | 19 +-
.../contrib/hexagon/profiling/process_lwp_data.py | 36 +-
python/tvm/contrib/hexagon/tools.py | 42 +-
python/tvm/contrib/hipblas.py | 1 +
python/tvm/contrib/miopen.py | 4 +-
python/tvm/contrib/mkl.py | 1 +
python/tvm/contrib/mps.py | 2 +-
python/tvm/contrib/mrvl.py | 5 +-
python/tvm/contrib/msc/core/codegen/codegen.py | 20 +-
python/tvm/contrib/msc/core/frontend/translate.py | 20 +-
python/tvm/contrib/msc/core/gym/__init__.py | 1 +
.../tvm/contrib/msc/core/gym/agent/base_agent.py | 19 +-
python/tvm/contrib/msc/core/gym/agent/method.py | 5 +-
.../tvm/contrib/msc/core/gym/agent/search_agent.py | 2 +
.../tvm/contrib/msc/core/gym/control/__init__.py | 1 +
.../tvm/contrib/msc/core/gym/control/configer.py | 2 +-
.../tvm/contrib/msc/core/gym/control/controller.py | 8 +-
.../tvm/contrib/msc/core/gym/control/namespace.py | 4 +-
python/tvm/contrib/msc/core/gym/control/service.py | 26 +-
python/tvm/contrib/msc/core/gym/control/worker.py | 15 +-
.../contrib/msc/core/gym/environment/base_env.py | 19 +-
.../tvm/contrib/msc/core/gym/environment/method.py | 7 +-
.../contrib/msc/core/gym/environment/prune_env.py | 4 +-
.../msc/core/gym/environment/quantize_env.py | 4 +-
python/tvm/contrib/msc/core/gym/namespace.py | 4 +-
python/tvm/contrib/msc/core/ir/graph.py | 5 +-
python/tvm/contrib/msc/core/runtime/__init__.py | 1 +
python/tvm/contrib/msc/core/runtime/hook.py | 12 +-
python/tvm/contrib/msc/core/runtime/jit.py | 9 +-
python/tvm/contrib/msc/core/runtime/runner.py | 28 +-
python/tvm/contrib/msc/core/tools/__init__.py | 1 +
python/tvm/contrib/msc/core/tools/configer.py | 4 +-
.../tvm/contrib/msc/core/tools/distill/__init__.py | 1 +
.../tvm/contrib/msc/core/tools/distill/configer.py | 4 +-
.../contrib/msc/core/tools/distill/distiller.py | 20 +-
.../tvm/contrib/msc/core/tools/distill/method.py | 7 +-
python/tvm/contrib/msc/core/tools/execute.py | 17 +-
.../tvm/contrib/msc/core/tools/prune/__init__.py | 1 +
.../tvm/contrib/msc/core/tools/prune/configer.py | 5 +-
python/tvm/contrib/msc/core/tools/prune/method.py | 7 +-
python/tvm/contrib/msc/core/tools/prune/pruner.py | 32 +-
.../contrib/msc/core/tools/quantize/__init__.py | 1 +
.../contrib/msc/core/tools/quantize/configer.py | 5 +-
.../tvm/contrib/msc/core/tools/quantize/method.py | 10 +-
.../contrib/msc/core/tools/quantize/quantizer.py | 8 +-
python/tvm/contrib/msc/core/tools/tool.py | 111 +-
.../tvm/contrib/msc/core/tools/track/__init__.py | 1 +
.../tvm/contrib/msc/core/tools/track/configer.py | 4 +-
python/tvm/contrib/msc/core/tools/track/method.py | 9 +-
python/tvm/contrib/msc/core/tools/track/tracker.py | 5 +-
python/tvm/contrib/msc/core/transform/pattern.py | 59 +-
python/tvm/contrib/msc/core/transform/transform.py | 2 +-
python/tvm/contrib/msc/core/utils/__init__.py | 1 +
python/tvm/contrib/msc/core/utils/arguments.py | 17 +-
python/tvm/contrib/msc/core/utils/dataset.py | 46 +-
python/tvm/contrib/msc/core/utils/expr.py | 8 +-
python/tvm/contrib/msc/core/utils/file.py | 32 +-
python/tvm/contrib/msc/core/utils/info.py | 38 +-
python/tvm/contrib/msc/core/utils/log.py | 22 +-
python/tvm/contrib/msc/core/utils/message.py | 12 +-
python/tvm/contrib/msc/core/utils/namespace.py | 2 +-
python/tvm/contrib/msc/core/utils/register.py | 17 +-
.../msc/framework/tensorflow/codegen/codegen.py | 9 +-
.../msc/framework/tensorflow/frontend/translate.py | 3 +-
.../msc/framework/tensorflow/runtime/runner.py | 13 +-
.../msc/framework/tensorflow/tools/__init__.py | 1 +
.../tensorflow/tools/distill/distiller.py | 6 +-
.../msc/framework/tensorflow/tools/prune/pruner.py | 6 +-
.../tensorflow/tools/quantize/quantizer.py | 6 +-
.../framework/tensorflow/tools/track/tracker.py | 4 +-
.../msc/framework/tensorrt/codegen/codegen.py | 16 +-
.../msc/framework/tensorrt/codegen/utils.py | 7 +-
.../msc/framework/tensorrt/frontend/translate.py | 4 +-
.../msc/framework/tensorrt/runtime/runner.py | 8 +-
.../msc/framework/tensorrt/tools/__init__.py | 1 +
.../framework/tensorrt/tools/distill/distiller.py | 6 +-
.../msc/framework/tensorrt/tools/prune/pruner.py | 6 +-
.../framework/tensorrt/tools/quantize/__init__.py | 1 +
.../framework/tensorrt/tools/quantize/method.py | 6 +-
.../framework/tensorrt/tools/quantize/quantizer.py | 30 +-
.../msc/framework/tensorrt/tools/track/tracker.py | 6 +-
.../msc/framework/tensorrt/transform/pattern.py | 14 +-
.../msc/framework/tensorrt/transform/transform.py | 4 +-
.../contrib/msc/framework/torch/codegen/codegen.py | 7 +-
.../msc/framework/torch/frontend/translate.py | 7 +-
.../msc/framework/torch/runtime/__init__.py | 1 +
.../tvm/contrib/msc/framework/torch/runtime/jit.py | 7 +-
.../contrib/msc/framework/torch/runtime/runner.py | 28 +-
.../contrib/msc/framework/torch/tools/__init__.py | 1 +
.../msc/framework/torch/tools/distill/distiller.py | 11 +-
.../msc/framework/torch/tools/distill/method.py | 5 +-
.../msc/framework/torch/tools/prune/pruner.py | 6 +-
.../msc/framework/torch/tools/quantize/__init__.py | 1 +
.../msc/framework/torch/tools/quantize/method.py | 7 +-
.../framework/torch/tools/quantize/quantizer.py | 6 +-
.../msc/framework/torch/tools/track/tracker.py | 4 +-
.../contrib/msc/framework/tvm/codegen/codegen.py | 4 +-
.../contrib/msc/framework/tvm/runtime/runner.py | 21 +-
.../contrib/msc/framework/tvm/tools/__init__.py | 1 +
.../msc/framework/tvm/tools/distill/distiller.py | 6 +-
.../msc/framework/tvm/tools/prune/pruner.py | 6 +-
.../msc/framework/tvm/tools/quantize/__init__.py | 1 +
.../msc/framework/tvm/tools/quantize/method.py | 9 +-
.../msc/framework/tvm/tools/quantize/quantizer.py | 6 +-
.../msc/framework/tvm/tools/track/tracker.py | 6 +-
python/tvm/contrib/msc/pipeline/dynamic.py | 5 +-
python/tvm/contrib/msc/pipeline/manager.py | 3 +-
python/tvm/contrib/msc/pipeline/pipeline.py | 36 +-
python/tvm/contrib/msc/pipeline/utils.py | 6 +-
python/tvm/contrib/msc/pipeline/worker.py | 47 +-
python/tvm/contrib/msc/pipeline/wrapper.py | 11 +-
python/tvm/contrib/msc/plugin/build.py | 25 +-
python/tvm/contrib/msc/plugin/codegen/codegen.py | 17 +-
python/tvm/contrib/msc/plugin/register.py | 2 +-
python/tvm/contrib/msc/plugin/utils.py | 3 +-
python/tvm/contrib/ndk.py | 16 +-
python/tvm/contrib/nnpack.py | 4 +-
python/tvm/contrib/pickle_memoize.py | 5 +-
python/tvm/contrib/popen_pool.py | 21 +-
python/tvm/contrib/random.py | 4 +-
python/tvm/contrib/rocblas.py | 1 +
python/tvm/contrib/rocm.py | 8 +-
python/tvm/contrib/rpc.py | 5 +-
python/tvm/contrib/spirv.py | 6 +-
python/tvm/contrib/tar.py | 5 +-
python/tvm/contrib/tflite_runtime.py | 4 +-
python/tvm/contrib/thrust.py | 1 +
python/tvm/contrib/tvmjs.py | 5 +-
python/tvm/contrib/utils.py | 9 +-
python/tvm/contrib/xcode.py | 5 +-
python/tvm/driver/__init__.py | 2 +
python/tvm/driver/_ffi_api.py | 1 +
python/tvm/driver/build_module.py | 1 +
python/tvm/error.py | 1 +
python/tvm/exec/disco_remote_socket_session.py | 4 +-
python/tvm/exec/disco_worker.py | 10 +-
python/tvm/exec/gpu_memory_bandwidth.py | 1 +
python/tvm/exec/popen_worker.py | 10 +-
python/tvm/exec/query_rpc_tracker.py | 4 +-
python/tvm/exec/rpc_proxy.py | 6 +-
python/tvm/exec/rpc_server.py | 6 +-
python/tvm/exec/rpc_tracker.py | 4 +-
python/tvm/ffi.py | 1 +
python/tvm/ir/__init__.py | 1 +
python/tvm/ir/_ffi_analysis_api.py | 1 -
python/tvm/ir/_ffi_api.py | 2 +-
python/tvm/ir/_ffi_instrument_api.py | 1 +
python/tvm/ir/_ffi_transform_api.py | 2 +-
python/tvm/ir/analysis.py | 1 +
python/tvm/ir/attrs.py | 4 +-
python/tvm/ir/base.py | 11 +-
python/tvm/ir/container.py | 1 +
python/tvm/ir/diagnostics/__init__.py | 2 +
python/tvm/ir/diagnostics/_ffi_api.py | 2 +-
python/tvm/ir/expr.py | 4 +-
python/tvm/ir/function.py | 8 +-
python/tvm/ir/global_info.py | 7 +-
python/tvm/ir/instrument.py | 1 +
python/tvm/ir/json_compact.py | 1 +
python/tvm/ir/module.py | 11 +-
python/tvm/ir/op.py | 1 +
python/tvm/ir/supply.py | 7 +-
python/tvm/ir/transform.py | 4 +-
python/tvm/ir/type.py | 7 +-
python/tvm/ir/type_relation.py | 3 +-
python/tvm/libinfo.py | 1 +
python/tvm/relax/__init__.py | 2 +
python/tvm/relax/_ffi_api.py | 1 +
python/tvm/relax/analysis/__init__.py | 3 +-
python/tvm/relax/analysis/_ffi_api.py | 1 +
python/tvm/relax/analysis/analysis.py | 12 +-
python/tvm/relax/analysis/estimate_memory_usage.py | 5 +-
python/tvm/relax/backend/__init__.py | 1 +
python/tvm/relax/backend/adreno/__init__.py | 1 +
python/tvm/relax/backend/adreno/clml.py | 8 +-
python/tvm/relax/backend/adreno/pipeline.py | 1 +
.../tvm/relax/backend/adreno/transform/__init__.py | 3 +-
.../tvm/relax/backend/adreno/transform/_ffi_api.py | 1 +
.../relax/backend/adreno/transform/transform.py | 1 +
python/tvm/relax/backend/contrib/nnapi.py | 8 +-
python/tvm/relax/backend/cpu_generic/__init__.py | 2 +
python/tvm/relax/backend/cpu_generic/pipeline.py | 1 +
python/tvm/relax/backend/cuda/__init__.py | 2 +
python/tvm/relax/backend/cuda/cublas.py | 1 +
python/tvm/relax/backend/cuda/cudnn.py | 1 +
python/tvm/relax/backend/cuda/cutlass.py | 3 +-
python/tvm/relax/backend/cuda/flashinfer.py | 1 +
python/tvm/relax/backend/cuda/pipeline.py | 1 +
python/tvm/relax/backend/dispatch_sampling.py | 1 -
python/tvm/relax/backend/gpu_generic/__init__.py | 2 +
python/tvm/relax/backend/gpu_generic/pipeline.py | 1 +
python/tvm/relax/backend/gpu_generic/sampling.py | 1 +
python/tvm/relax/backend/metal/coreml.py | 18 +-
python/tvm/relax/backend/patterns.py | 8 +-
python/tvm/relax/backend/rocm/__init__.py | 2 +
python/tvm/relax/backend/rocm/hipblas.py | 1 +
python/tvm/relax/backend/rocm/pipeline.py | 1 +
python/tvm/relax/backend/utils.py | 1 +
python/tvm/relax/base_py_module.py | 4 +-
python/tvm/relax/binding_rewrite.py | 11 +-
python/tvm/relax/block_builder.py | 15 +-
python/tvm/relax/distributed/__init__.py | 1 +
python/tvm/relax/distributed/_ffi_api.py | 1 +
python/tvm/relax/distributed/global_info.py | 8 +-
python/tvm/relax/distributed/struct_info.py | 15 +-
python/tvm/relax/distributed/transform/__init__.py | 3 +-
python/tvm/relax/distributed/transform/_ffi_api.py | 1 +
.../tvm/relax/distributed/transform/transform.py | 1 +
python/tvm/relax/dpl/__init__.py | 1 +
python/tvm/relax/dpl/_ffi.py | 1 +
python/tvm/relax/dpl/context.py | 3 +-
python/tvm/relax/dpl/pattern.py | 9 +-
python/tvm/relax/dpl/rewrite.py | 9 +-
python/tvm/relax/exec_builder.py | 12 +-
python/tvm/relax/expr.py | 47 +-
python/tvm/relax/expr_functor.py | 10 +-
python/tvm/relax/frontend/__init__.py | 2 +
python/tvm/relax/frontend/common.py | 2 +
python/tvm/relax/frontend/nn/__init__.py | 2 +
python/tvm/relax/frontend/nn/_tensor_op.py | 1 +
python/tvm/relax/frontend/nn/core.py | 3 +-
python/tvm/relax/frontend/nn/exporter.py | 1 +
python/tvm/relax/frontend/nn/extern.py | 15 +-
python/tvm/relax/frontend/nn/llm/__init__.py | 1 +
python/tvm/relax/frontend/nn/llm/kv_cache.py | 22 +-
python/tvm/relax/frontend/nn/llm/tree_attn.py | 7 +-
python/tvm/relax/frontend/nn/modules.py | 4 +-
python/tvm/relax/frontend/nn/op.py | 25 +-
python/tvm/relax/frontend/nn/spec.py | 1 +
python/tvm/relax/frontend/nn/torch.py | 6 +-
python/tvm/relax/frontend/nn/visitor.py | 1 +
python/tvm/relax/frontend/onnx/__init__.py | 2 +
python/tvm/relax/frontend/onnx/onnx_frontend.py | 125 +--
python/tvm/relax/frontend/stablehlo/__init__.py | 2 +
.../frontend/stablehlo/stablehlo_translator.py | 11 +-
python/tvm/relax/frontend/torch/__init__.py | 2 +
.../frontend/torch/base_fx_graph_translator.py | 25 +-
python/tvm/relax/frontend/torch/dynamo.py | 2 +-
.../frontend/torch/exported_program_translator.py | 2 +
python/tvm/relax/frontend/torch/fx_translator.py | 14 +-
python/tvm/relax/ir/instrument.py | 1 +
python/tvm/relax/op/__init__.py | 1 +
python/tvm/relax/op/_ffi_api.py | 1 +
python/tvm/relax/op/_op_gradient.py | 31 +-
python/tvm/relax/op/base.py | 12 +-
python/tvm/relax/op/binary.py | 3 +-
python/tvm/relax/op/builtin/_ffi_api.py | 1 +
python/tvm/relax/op/ccl/__init__.py | 2 +
python/tvm/relax/op/ccl/_ffi_api.py | 1 +
python/tvm/relax/op/ccl/ccl.py | 2 +-
python/tvm/relax/op/create.py | 7 +-
python/tvm/relax/op/datatype.py | 3 +-
python/tvm/relax/op/distributed/__init__.py | 1 +
python/tvm/relax/op/distributed/_ffi_api.py | 1 +
python/tvm/relax/op/distributed/distributed.py | 13 +-
python/tvm/relax/op/grad/_ffi_api.py | 1 +
python/tvm/relax/op/grad/grad.py | 3 +-
python/tvm/relax/op/image/__init__.py | 2 +
python/tvm/relax/op/image/_ffi_api.py | 1 +
python/tvm/relax/op/image/image.py | 4 +-
python/tvm/relax/op/index.py | 5 +-
python/tvm/relax/op/linear_algebra.py | 4 +-
python/tvm/relax/op/manipulate.py | 9 +-
python/tvm/relax/op/mask.py | 3 +-
python/tvm/relax/op/memory/__init__.py | 1 +
python/tvm/relax/op/memory/_ffi_api.py | 1 +
python/tvm/relax/op/memory/memory.py | 5 +-
python/tvm/relax/op/memory/view.py | 4 +-
python/tvm/relax/op/nn/__init__.py | 2 +
python/tvm/relax/op/nn/_ffi_api.py | 1 +
python/tvm/relax/op/nn/nn.py | 9 +-
python/tvm/relax/op/op_attrs.py | 4 +-
python/tvm/relax/op/search.py | 3 +-
python/tvm/relax/op/set.py | 4 +-
python/tvm/relax/op/sorting.py | 3 +-
python/tvm/relax/op/statistical.py | 4 +-
python/tvm/relax/op/ternary.py | 3 +-
python/tvm/relax/op/unary.py | 3 +-
python/tvm/relax/op/vision/__init__.py | 2 +
python/tvm/relax/op/vision/_ffi_api.py | 1 +
python/tvm/relax/op/vision/nms.py | 1 +
python/tvm/relax/op/vm/_ffi_api.py | 1 +
python/tvm/relax/op/vm/vm.py | 5 +-
python/tvm/relax/pipeline.py | 5 +-
python/tvm/relax/relax_to_pyfunc_converter.py | 3 +-
python/tvm/relax/struct_info.py | 37 +-
python/tvm/relax/testing/__init__.py | 1 +
python/tvm/relax/testing/ast_printer.py | 4 +-
python/tvm/relax/testing/attention.py | 4 +-
python/tvm/relax/testing/lib_comparator.py | 3 +-
python/tvm/relax/testing/matmul.py | 1 +
python/tvm/relax/testing/nn.py | 5 +-
python/tvm/relax/testing/runtime_builtin.py | 1 +
python/tvm/relax/testing/transform.py | 8 +-
python/tvm/relax/testing/vm.py | 4 +-
python/tvm/relax/training/__init__.py | 1 +
python/tvm/relax/training/_ffi_api.py | 1 +
python/tvm/relax/training/loss.py | 7 +-
python/tvm/relax/training/optimizer.py | 7 +-
python/tvm/relax/training/setup_trainer.py | 7 +-
python/tvm/relax/training/trainer.py | 6 +-
python/tvm/relax/training/utils.py | 5 +-
python/tvm/relax/transform/__init__.py | 1 +
python/tvm/relax/transform/_ffi_api.py | 1 +
.../tvm/relax/transform/attach_external_modules.py | 1 +
python/tvm/relax/transform/fast_math.py | 1 +
.../fold_batch_norm_to_conv2d_for_inference.py | 5 +-
.../tvm/relax/transform/lazy_transform_params.py | 1 +
.../tvm/relax/transform/legalize_ops/__init__.py | 2 +
.../transform/legalize_ops/adreno/__init__.py | 2 +
.../transform/legalize_ops/adreno/convolution.py | 3 +-
python/tvm/relax/transform/legalize_ops/binary.py | 4 +-
python/tvm/relax/transform/legalize_ops/ccl.py | 18 +-
python/tvm/relax/transform/legalize_ops/common.py | 5 +-
python/tvm/relax/transform/legalize_ops/create.py | 1 +
.../tvm/relax/transform/legalize_ops/datatype.py | 4 +-
.../relax/transform/legalize_ops/distributed.py | 4 +-
python/tvm/relax/transform/legalize_ops/grad.py | 2 +
python/tvm/relax/transform/legalize_ops/image.py | 2 +
python/tvm/relax/transform/legalize_ops/index.py | 8 +-
.../tvm/relax/transform/legalize_ops/inspect_op.py | 8 +-
.../relax/transform/legalize_ops/linear_algebra.py | 6 +-
.../tvm/relax/transform/legalize_ops/manipulate.py | 13 +-
python/tvm/relax/transform/legalize_ops/nn.py | 13 +-
python/tvm/relax/transform/legalize_ops/qdq.py | 4 +-
python/tvm/relax/transform/legalize_ops/search.py | 5 +-
.../relax/transform/legalize_ops/statistical.py | 9 +-
python/tvm/relax/transform/legalize_ops/unary.py | 3 +-
python/tvm/relax/transform/legalize_ops/vision.py | 1 +
.../relax/transform/optimize_layout_transform.py | 1 +
.../relax/transform/remove_redundant_reshape.py | 1 +
python/tvm/relax/transform/transform.py | 16 +-
python/tvm/relax/ty.py | 4 +-
python/tvm/relax/type_converter.py | 3 +-
python/tvm/relax/utils.py | 34 +-
python/tvm/relax/vm_build.py | 3 +-
python/tvm/rpc/__init__.py | 1 +
python/tvm/rpc/_ffi_api.py | 2 +-
python/tvm/rpc/base.py | 22 +-
python/tvm/rpc/client.py | 19 +-
python/tvm/rpc/minrpc.py | 3 +
python/tvm/rpc/proxy.py | 44 +-
python/tvm/rpc/server.py | 45 +-
python/tvm/rpc/server_ios_launcher.py | 9 +-
python/tvm/rpc/testing.py | 4 +-
python/tvm/rpc/tornado_util.py | 13 +-
python/tvm/rpc/tracker.py | 26 +-
python/tvm/runtime/__init__.py | 1 +
python/tvm/runtime/_ffi_api.py | 1 +
python/tvm/runtime/_ffi_node_api.py | 1 +
python/tvm/runtime/_tensor.py | 4 +-
python/tvm/runtime/container.py | 4 +-
python/tvm/runtime/device.py | 2 +-
python/tvm/runtime/disco/__init__.py | 2 +
python/tvm/runtime/disco/_ffi_api.py | 1 +
python/tvm/runtime/disco/process_pool.py | 2 +
python/tvm/runtime/disco/session.py | 6 +-
python/tvm/runtime/executable.py | 5 +-
python/tvm/runtime/module.py | 25 +-
python/tvm/runtime/object.py | 4 +-
python/tvm/runtime/object_generic.py | 2 +
python/tvm/runtime/packed_func.py | 1 +
python/tvm/runtime/params.py | 3 +-
python/tvm/runtime/profiling/__init__.py | 1 +
python/tvm/runtime/profiling/_ffi_api.py | 1 +
python/tvm/runtime/script_printer.py | 5 +-
python/tvm/runtime/support.py | 16 +-
python/tvm/runtime/vm.py | 9 +-
python/tvm/s_tir/__init__.py | 1 +
python/tvm/s_tir/_ffi_api.py | 2 +-
python/tvm/s_tir/analysis/__init__.py | 2 +
python/tvm/s_tir/analysis/_ffi_api.py | 2 +-
python/tvm/s_tir/backend/adreno/__init__.py | 1 +
python/tvm/s_tir/backend/adreno/pipeline.py | 2 +-
.../tvm/s_tir/backend/adreno/transform/__init__.py | 3 +-
.../tvm/s_tir/backend/adreno/transform/_ffi_api.py | 2 +-
python/tvm/s_tir/data_layout.py | 2 +
python/tvm/s_tir/dlight/__init__.py | 2 +
python/tvm/s_tir/dlight/adreno/__init__.py | 2 +
python/tvm/s_tir/dlight/adreno/convolution.py | 5 +-
python/tvm/s_tir/dlight/adreno/fallback.py | 21 +-
python/tvm/s_tir/dlight/adreno/layout_transform.py | 22 +-
python/tvm/s_tir/dlight/adreno/pool.py | 6 +-
python/tvm/s_tir/dlight/adreno/utils.py | 2 +-
python/tvm/s_tir/dlight/analysis/__init__.py | 2 +
.../tvm/s_tir/dlight/analysis/common_analysis.py | 14 +-
python/tvm/s_tir/dlight/analysis/gemv.py | 3 +-
python/tvm/s_tir/dlight/base/__init__.py | 2 +
python/tvm/s_tir/dlight/base/common_schedules.py | 1 +
python/tvm/s_tir/dlight/base/schedule_rule.py | 3 +-
python/tvm/s_tir/dlight/base/transform.py | 3 +-
python/tvm/s_tir/dlight/base/utils.py | 3 +-
python/tvm/s_tir/dlight/benchmark/__init__.py | 2 +
python/tvm/s_tir/dlight/benchmark/bench.py | 9 +-
python/tvm/s_tir/dlight/benchmark/extract.py | 6 +-
python/tvm/s_tir/dlight/benchmark/utils.py | 2 +-
python/tvm/s_tir/dlight/cpu/__init__.py | 2 +
python/tvm/s_tir/dlight/cpu/gemv.py | 3 +-
python/tvm/s_tir/dlight/gpu/__init__.py | 2 +
python/tvm/s_tir/dlight/gpu/fallback.py | 3 +-
python/tvm/s_tir/dlight/gpu/gemv.py | 3 +-
python/tvm/s_tir/dlight/gpu/general_reduction.py | 3 +-
python/tvm/s_tir/dlight/gpu/low_batch_gemv.py | 3 +-
python/tvm/s_tir/dlight/gpu/matmul.py | 7 +-
python/tvm/s_tir/dlight/gpu/reduction.py | 3 +-
python/tvm/s_tir/dlight/gpu/rmsnorm.py | 3 +-
python/tvm/s_tir/dlight/gpu/transpose.py | 5 +-
python/tvm/s_tir/meta_schedule/__init__.py | 2 +
python/tvm/s_tir/meta_schedule/_ffi_api.py | 1 +
python/tvm/s_tir/meta_schedule/arg_info.py | 2 +
python/tvm/s_tir/meta_schedule/builder/__init__.py | 2 +
python/tvm/s_tir/meta_schedule/builder/builder.py | 4 +-
.../s_tir/meta_schedule/builder/local_builder.py | 4 +-
.../tvm/s_tir/meta_schedule/cost_model/__init__.py | 2 +
.../s_tir/meta_schedule/cost_model/cost_model.py | 8 +-
.../tvm/s_tir/meta_schedule/cost_model/metric.py | 1 +
.../s_tir/meta_schedule/cost_model/mlp_model.py | 2 +
.../s_tir/meta_schedule/cost_model/random_model.py | 5 +-
.../s_tir/meta_schedule/cost_model/xgb_model.py | 10 +-
.../tvm/s_tir/meta_schedule/database/__init__.py | 2 +
.../tvm/s_tir/meta_schedule/database/database.py | 19 +-
.../s_tir/meta_schedule/database/json_database.py | 1 +
.../meta_schedule/database/memory_database.py | 1 +
.../database/ordered_union_database.py | 1 +
.../meta_schedule/database/schedule_fn_database.py | 2 +
.../s_tir/meta_schedule/database/union_database.py | 1 +
python/tvm/s_tir/meta_schedule/extracted_task.py | 2 +
.../meta_schedule/feature_extractor/__init__.py | 2 +
.../feature_extractor/feature_extractor.py | 2 +
.../feature_extractor/per_store_feature.py | 1 +
.../feature_extractor/random_feature_extractor.py | 2 +
python/tvm/s_tir/meta_schedule/logging.py | 1 +
.../meta_schedule/measure_callback/__init__.py | 2 +
.../measure_callback/add_to_database.py | 1 +
.../measure_callback/measure_callback.py | 1 +
.../measure_callback/remove_build_artifact.py | 1 +
.../measure_callback/update_cost_model.py | 1 +
python/tvm/s_tir/meta_schedule/mutator/__init__.py | 2 +
.../mutator/mutate_compute_location.py | 1 +
.../s_tir/meta_schedule/mutator/mutate_parallel.py | 1 +
.../meta_schedule/mutator/mutate_thread_binding.py | 1 +
.../meta_schedule/mutator/mutate_tile_size.py | 1 +
.../s_tir/meta_schedule/mutator/mutate_unroll.py | 1 +
python/tvm/s_tir/meta_schedule/mutator/mutator.py | 2 +
.../meta_schedule/post_optimization/__init__.py | 2 +
.../meta_schedule/post_optimization/droplet.py | 5 +-
.../meta_schedule/post_optimization/post_opt.py | 3 +-
.../s_tir/meta_schedule/post_optimization/space.py | 9 +-
.../s_tir/meta_schedule/post_optimization/utils.py | 9 +-
.../tvm/s_tir/meta_schedule/postproc/__init__.py | 2 +
.../postproc/disallow_async_strided_mem_copy.py | 1 +
.../postproc/disallow_dynamic_loop.py | 1 +
.../tvm/s_tir/meta_schedule/postproc/postproc.py | 2 +
.../postproc/rewrite_cooperative_fetch.py | 1 +
.../postproc/rewrite_parallel_vectorize_unroll.py | 1 +
.../postproc/rewrite_reduction_block.py | 1 +
.../meta_schedule/postproc/rewrite_tensorize.py | 1 +
.../meta_schedule/postproc/verify_gpu_code.py | 1 +
.../meta_schedule/postproc/verify_vtcm_limit.py | 1 +
python/tvm/s_tir/meta_schedule/profiler.py | 2 +
.../tvm/s_tir/meta_schedule/relax_integration.py | 6 +-
python/tvm/s_tir/meta_schedule/runner/__init__.py | 2 +
python/tvm/s_tir/meta_schedule/runner/config.py | 1 +
.../tvm/s_tir/meta_schedule/runner/local_runner.py | 3 +-
.../tvm/s_tir/meta_schedule/runner/rpc_runner.py | 1 +
python/tvm/s_tir/meta_schedule/runner/runner.py | 2 +
python/tvm/s_tir/meta_schedule/runner/utils.py | 2 +
.../tvm/s_tir/meta_schedule/schedule/__init__.py | 2 +
.../schedule/cuda/layout_transform.py | 8 +-
.../s_tir/meta_schedule/schedule_rule/__init__.py | 2 +
.../meta_schedule/schedule_rule/add_rfactor.py | 1 +
.../schedule_rule/apply_custom_rule.py | 1 +
.../s_tir/meta_schedule/schedule_rule/auto_bind.py | 1 +
.../meta_schedule/schedule_rule/auto_inline.py | 1 +
.../schedule_rule/cross_thread_reduction.py | 1 +
.../schedule_rule/multi_level_tiling.py | 6 +-
.../schedule_rule/parallel_vectorize_unroll.py | 1 +
.../schedule_rule/random_compute_location.py | 1 +
.../meta_schedule/schedule_rule/schedule_rule.py | 2 +
.../search_strategy/evolutionary_search.py | 1 +
.../meta_schedule/search_strategy/replay_func.py | 1 +
.../meta_schedule/search_strategy/replay_trace.py | 1 +
.../search_strategy/search_strategy.py | 2 +
.../meta_schedule/space_generator/__init__.py | 2 +
.../space_generator/post_order_apply.py | 1 +
.../meta_schedule/space_generator/schedule_fn.py | 1 +
.../space_generator/space_generator.py | 2 +
.../space_generator/space_generator_union.py | 1 +
.../s_tir/meta_schedule/task_scheduler/__init__.py | 2 +
.../meta_schedule/task_scheduler/gradient_based.py | 1 +
.../meta_schedule/task_scheduler/round_robin.py | 1 +
.../meta_schedule/task_scheduler/task_scheduler.py | 5 +-
.../meta_schedule/testing/custom_builder_runner.py | 5 +-
.../testing/distributed_measure_candidates.py | 1 +
.../s_tir/meta_schedule/testing/dummy_object.py | 1 +
.../tvm/s_tir/meta_schedule/testing/local_rpc.py | 3 +-
.../meta_schedule/testing/space_generation.py | 4 +-
.../tvm/s_tir/meta_schedule/testing/te_workload.py | 1 +
.../tvm/s_tir/meta_schedule/testing/tune_utils.py | 4 +-
.../meta_schedule/testing/validate_database.py | 29 +-
python/tvm/s_tir/meta_schedule/tir_integration.py | 1 +
python/tvm/s_tir/meta_schedule/trace_apply.py | 2 +
python/tvm/s_tir/meta_schedule/tune.py | 3 +-
python/tvm/s_tir/meta_schedule/tune_context.py | 17 +-
python/tvm/s_tir/meta_schedule/utils.py | 18 +-
python/tvm/s_tir/pipeline.py | 2 +-
python/tvm/s_tir/sblock_dependence_info.py | 8 +-
python/tvm/s_tir/sblock_scope.py | 4 +-
python/tvm/s_tir/schedule/__init__.py | 1 +
python/tvm/s_tir/schedule/_ffi_api.py | 1 +
python/tvm/s_tir/schedule/_type_checker.py | 3 +-
python/tvm/s_tir/schedule/analysis.py | 7 +-
python/tvm/s_tir/schedule/instruction.py | 2 +
python/tvm/s_tir/schedule/schedule.py | 16 +-
python/tvm/s_tir/schedule/state.py | 6 +-
python/tvm/s_tir/schedule/testing.py | 3 +-
python/tvm/s_tir/schedule/trace.py | 2 +
python/tvm/s_tir/schedule/transform.py | 4 +-
python/tvm/s_tir/tensor_intrin/__init__.py | 2 +
python/tvm/s_tir/tensor_intrin/arm_cpu.py | 3 +-
python/tvm/s_tir/tensor_intrin/cuda.py | 20 +-
.../tvm/s_tir/tensor_intrin/dot_product_common.py | 2 +
python/tvm/s_tir/tensor_intrin/hexagon.py | 2 +
python/tvm/s_tir/tensor_intrin/metal.py | 1 +
python/tvm/s_tir/tensor_intrin/riscv_cpu.py | 5 +-
python/tvm/s_tir/tensor_intrin/rocm.py | 6 +-
python/tvm/s_tir/tensor_intrin/x86.py | 3 +-
python/tvm/s_tir/transform/__init__.py | 1 +
python/tvm/s_tir/transform/_ffi_api.py | 2 +-
python/tvm/s_tir/transform/transform.py | 4 +-
python/tvm/script/__init__.py | 2 +
python/tvm/script/_ffi_api.py | 19 +-
python/tvm/script/highlight.py | 3 +-
python/tvm/script/ir_builder/__init__.py | 2 +
python/tvm/script/ir_builder/_ffi_api.py | 1 +
python/tvm/script/ir_builder/base.py | 2 +
python/tvm/script/ir_builder/ir/__init__.py | 2 +
python/tvm/script/ir_builder/ir/_ffi_api.py | 1 +
python/tvm/script/ir_builder/ir/frame.py | 3 +-
python/tvm/script/ir_builder/ir/ir.py | 3 +-
python/tvm/script/ir_builder/relax/__init__.py | 2 +
python/tvm/script/ir_builder/relax/_ffi_api.py | 1 +
.../ir_builder/relax/distributed/__init__.py | 2 +
.../ir_builder/relax/distributed/_ffi_api.py | 5 +-
.../tvm/script/ir_builder/relax/distributed/ir.py | 29 +-
python/tvm/script/ir_builder/relax/frame.py | 13 +-
python/tvm/script/ir_builder/relax/ir.py | 76 +-
python/tvm/script/ir_builder/tir/__init__.py | 2 +
python/tvm/script/ir_builder/tir/_ffi_api.py | 1 +
.../tvm/script/ir_builder/tir/external_kernel.py | 7 +-
python/tvm/script/ir_builder/tir/frame.py | 32 +-
python/tvm/script/ir_builder/tir/ir.py | 319 +++---
python/tvm/script/ir_builder/tir/triton.py | 6 +-
python/tvm/script/ir_builder/tir/utils.py | 3 +-
python/tvm/script/parser/__init__.py | 2 +
python/tvm/script/parser/_core.py | 1 +
python/tvm/script/parser/core/__init__.py | 2 +
python/tvm/script/parser/core/doc_core.py | 154 +--
python/tvm/script/parser/core/entry.py | 12 +-
python/tvm/script/parser/core/error.py | 1 +
python/tvm/script/parser/core/parser.py | 4 +-
python/tvm/script/parser/core/utils.py | 3 +-
python/tvm/script/parser/ir/__init__.py | 8 +-
python/tvm/script/parser/ir/entry.py | 5 +-
python/tvm/script/parser/relax/__init__.py | 1 +
python/tvm/script/parser/relax/dist.py | 15 +-
python/tvm/script/parser/relax/entry.py | 11 +-
python/tvm/script/parser/relax/parser.py | 6 +-
python/tvm/script/parser/tir/entry.py | 1 +
python/tvm/script/parser/tir/parser.py | 1 -
python/tvm/script/printer/_ffi_api.py | 1 +
python/tvm/script/printer/doc.py | 7 +-
python/tvm/script/printer/doc_printer.py | 1 +
python/tvm/script/relax.py | 1 +
python/tvm/script/tir.py | 1 +
python/tvm/support.py | 13 +-
python/tvm/target/__init__.py | 4 +-
python/tvm/target/_ffi_api.py | 2 +-
python/tvm/target/codegen.py | 3 +-
python/tvm/target/datatype.py | 13 +-
python/tvm/target/detect_target.py | 2 +
python/tvm/target/intrin.py | 1 +
python/tvm/target/tag.py | 1 +
python/tvm/target/tag_registry/__init__.py | 18 +-
python/tvm/target/tag_registry/adreno.py | 1 +
python/tvm/target/tag_registry/arm_cpu.py | 2 +-
python/tvm/target/tag_registry/aws_cpu.py | 1 +
python/tvm/target/tag_registry/cuda.py | 1 +
python/tvm/target/tag_registry/hexagon.py | 1 +
python/tvm/target/tag_registry/metal.py | 1 +
python/tvm/target/tag_registry/registry.py | 2 +
python/tvm/target/tag_registry/riscv_cpu.py | 1 +
python/tvm/target/target.py | 7 +-
python/tvm/target/x86.py | 2 +
python/tvm/te/__init__.py | 5 +-
python/tvm/te/_ffi_api.py | 2 +-
python/tvm/te/operation.py | 1 +
python/tvm/te/tag.py | 5 +-
python/tvm/te/tensor.py | 4 +-
python/tvm/testing/__init__.py | 2 +
python/tvm/testing/_ffi_api.py | 2 +-
python/tvm/testing/attrs.py | 2 +
python/tvm/testing/plugin.py | 2 +-
python/tvm/testing/popen_pool.py | 2 +
python/tvm/testing/runner.py | 5 +-
python/tvm/testing/utils.py | 79 +-
python/tvm/tir/__init__.py | 2 +
python/tvm/tir/_ffi_api.py | 2 +-
python/tvm/tir/analysis/_ffi_api.py | 2 +-
python/tvm/tir/analysis/analysis.py | 1 +
python/tvm/tir/buffer.py | 9 +-
python/tvm/tir/build.py | 1 +
python/tvm/tir/expr.py | 79 +-
python/tvm/tir/function.py | 4 +-
python/tvm/tir/functor.py | 6 +-
python/tvm/tir/generic.py | 1 +
python/tvm/tir/op.py | 3 +-
python/tvm/tir/pipeline.py | 2 +-
python/tvm/tir/stmt.py | 40 +-
python/tvm/tir/stmt_functor.py | 1 +
python/tvm/tir/transform/__init__.py | 1 +
python/tvm/tir/transform/_ffi_api.py | 2 +-
python/tvm/tir/transform/function_pass.py | 8 +-
python/tvm/tir/transform/transform.py | 5 +-
python/tvm/topi/__init__.py | 2 +
python/tvm/topi/broadcast.py | 2 +
python/tvm/topi/cpp/__init__.py | 2 +
python/tvm/topi/cpp/cuda.py | 1 +
python/tvm/topi/cpp/generic.py | 1 +
python/tvm/topi/cpp/impl.py | 1 +
python/tvm/topi/cpp/nn.py | 1 +
python/tvm/topi/cpp/rocm.py | 1 +
python/tvm/topi/cpp/utils.py | 1 +
python/tvm/topi/cpp/vision/__init__.py | 2 +
python/tvm/topi/cpp/vision/yolo.py | 1 +
python/tvm/topi/cpp/x86.py | 1 +
python/tvm/topi/einsum.py | 1 +
python/tvm/topi/generic_op_impl.py | 2 +
python/tvm/topi/gpu/__init__.py | 2 +
python/tvm/topi/gpu/scan.py | 1 +
python/tvm/topi/gpu/sort.py | 11 +-
python/tvm/topi/image/__init__.py | 2 +
python/tvm/topi/image/dilation2d.py | 15 +-
python/tvm/topi/image/grid_sample.py | 7 +-
python/tvm/topi/image/resize.py | 3 +-
python/tvm/topi/index_put.py | 5 +-
python/tvm/topi/math.py | 1 +
python/tvm/topi/nn/__init__.py | 2 +
python/tvm/topi/nn/batch_matmul.py | 1 +
python/tvm/topi/nn/batch_norm.py | 4 +-
python/tvm/topi/nn/batch_to_space_nd.py | 2 +-
python/tvm/topi/nn/bitserial_conv2d.py | 6 +-
python/tvm/topi/nn/bitserial_dense.py | 3 +-
python/tvm/topi/nn/bitserial_util.py | 3 +
python/tvm/topi/nn/bnn.py | 11 +-
python/tvm/topi/nn/conv1d.py | 1 +
python/tvm/topi/nn/conv1d_transpose.py | 10 +-
python/tvm/topi/nn/conv2d.py | 8 +-
python/tvm/topi/nn/conv2d_transpose.py | 13 +-
python/tvm/topi/nn/conv3d.py | 1 +
python/tvm/topi/nn/conv3d_transpose.py | 1 +
python/tvm/topi/nn/correlation.py | 3 +-
python/tvm/topi/nn/deformable_conv2d.py | 5 +-
python/tvm/topi/nn/dense.py | 15 +-
python/tvm/topi/nn/depth_to_space.py | 3 +-
python/tvm/topi/nn/depthwise_conv2d.py | 12 +-
python/tvm/topi/nn/dilate.py | 5 +-
python/tvm/topi/nn/elemwise.py | 3 +
python/tvm/topi/nn/fifo_buffer.py | 3 +
python/tvm/topi/nn/flatten.py | 3 +-
python/tvm/topi/nn/group_norm.py | 1 +
python/tvm/topi/nn/instance_norm.py | 1 +
python/tvm/topi/nn/layer_norm.py | 1 +
python/tvm/topi/nn/local_response_norm.py | 2 +-
python/tvm/topi/nn/loss.py | 2 +-
python/tvm/topi/nn/lstm.py | 18 +-
python/tvm/topi/nn/mapping.py | 3 +
python/tvm/topi/nn/pad.py | 3 +-
python/tvm/topi/nn/pixel_shuffle.py | 1 -
python/tvm/topi/nn/pooling.py | 1 -
python/tvm/topi/nn/qnn.py | 1 +
python/tvm/topi/nn/rms_norm.py | 1 +
python/tvm/topi/nn/softmax.py | 2 +-
python/tvm/topi/nn/space_to_batch_nd.py | 2 +-
python/tvm/topi/nn/space_to_depth.py | 3 +-
python/tvm/topi/nn/upsampling.py | 5 +-
python/tvm/topi/nn/utils.py | 2 +-
python/tvm/topi/nn/winograd_util.py | 25 +-
python/tvm/topi/reduction.py | 2 +
python/tvm/topi/scan.py | 3 +-
python/tvm/topi/scatter.py | 11 +-
python/tvm/topi/scatter_elements.py | 5 +-
python/tvm/topi/searchsorted.py | 5 +-
python/tvm/topi/signal.py | 2 +
python/tvm/topi/slice_scatter.py | 2 +
python/tvm/topi/sort.py | 2 +
python/tvm/topi/sparse_reshape.py | 5 +-
python/tvm/topi/tensor.py | 1 -
python/tvm/topi/testing/__init__.py | 2 +
python/tvm/topi/testing/adaptive_pool_python.py | 1 +
python/tvm/topi/testing/attention_python.py | 9 +-
python/tvm/topi/testing/batch_matmul.py | 1 +
python/tvm/topi/testing/batch_norm.py | 1 +
python/tvm/topi/testing/batch_to_space_nd.py | 4 +-
python/tvm/topi/testing/conv1d_ncw_python.py | 2 +
.../topi/testing/conv1d_transpose_ncw_python.py | 2 +
.../topi/testing/conv2d_backcward_weight_python.py | 1 +
python/tvm/topi/testing/conv2d_hwcn_python.py | 2 +
python/tvm/topi/testing/conv2d_nchw_python.py | 1 +
python/tvm/topi/testing/conv2d_nhwc_python.py | 2 +
python/tvm/topi/testing/conv2d_transpose_python.py | 2 +
python/tvm/topi/testing/conv3d_ncdhw_python.py | 2 +
python/tvm/topi/testing/conv3d_ndhwc_python.py | 2 +
.../topi/testing/conv3d_transpose_ncdhw_python.py | 2 +
python/tvm/topi/testing/correlation_nchw_python.py | 1 +
python/tvm/topi/testing/crop_and_resize_python.py | 2 +
.../tvm/topi/testing/deformable_conv2d_python.py | 3 +
python/tvm/topi/testing/dense.py | 1 +
python/tvm/topi/testing/depth_to_space.py | 1 +
python/tvm/topi/testing/depthwise_conv2d_python.py | 2 +
python/tvm/topi/testing/dilate_python.py | 7 +-
python/tvm/topi/testing/gather_nd_python.py | 1 +
python/tvm/topi/testing/gather_python.py | 1 +
python/tvm/topi/testing/grid_sample_python.py | 2 +
python/tvm/topi/testing/group_norm_python.py | 1 +
python/tvm/topi/testing/instance_norm_python.py | 1 +
python/tvm/topi/testing/l2_normalize_python.py | 1 +
python/tvm/topi/testing/layer_norm_python.py | 1 +
python/tvm/topi/testing/lrn_python.py | 2 +
python/tvm/topi/testing/lstm_python.py | 1 +
python/tvm/topi/testing/matrix_set_diag.py | 1 +
python/tvm/topi/testing/nll_loss.py | 1 +
python/tvm/topi/testing/one_hot.py | 1 +
python/tvm/topi/testing/pool_grad_python.py | 1 +
python/tvm/topi/testing/poolnd_python.py | 4 +-
python/tvm/topi/testing/reorg_python.py | 1 +
python/tvm/topi/testing/resize_python.py | 3 +
python/tvm/topi/testing/rms_norm_python.py | 1 +
python/tvm/topi/testing/roi_align_python.py | 2 +
python/tvm/topi/testing/roi_pool_python.py | 2 +
python/tvm/topi/testing/searchsorted.py | 1 +
python/tvm/topi/testing/sequence_mask_python.py | 1 +
python/tvm/topi/testing/softmax_python.py | 1 +
python/tvm/topi/testing/space_to_batch_nd.py | 5 +-
python/tvm/topi/testing/space_to_depth.py | 1 +
python/tvm/topi/transform.py | 2 +
python/tvm/topi/unique.py | 1 +
python/tvm/topi/utils.py | 4 +-
python/tvm/topi/vision/__init__.py | 2 +
python/tvm/topi/vision/nms.py | 12 +-
python/tvm/topi/vision/nms_util.py | 1 +
src/ir/attr_functor.h | 6 +-
src/relax/transform/split_call_tir_by_pattern.cc | 2 +-
src/runtime/contrib/clml/clml_runtime.h | 6 +-
src/runtime/contrib/cublas/cublas_utils.h | 2 +-
src/runtime/contrib/cutlass/fp16_group_gemm.cuh | 3 +-
.../contrib/cutlass/fp8_groupwise_scaled_gemm.cuh | 9 +-
src/runtime/contrib/hipblas/hipblas_utils.h | 2 +-
src/runtime/contrib/random/random.cc | 24 +-
src/runtime/contrib/tensorrt/tensorrt_ops.cc | 28 +-
src/runtime/contrib/tflite/tflite_runtime.cc | 44 +-
src/runtime/contrib/vllm/attention_kernels.cu | 23 +-
src/runtime/contrib/vllm/attention_utils.cuh | 37 +-
src/runtime/contrib/vllm/dtype_float16.h | 21 +-
src/runtime/hexagon/hexagon_device_api.h | 2 +-
src/runtime/hexagon/ops/conv2d.h | 4 +-
src/runtime/texture.h | 2 +-
src/s_tir/schedule/ir_comparator.cc | 2 +-
src/s_tir/schedule/ir_comparator.h | 2 +-
src/s_tir/transform/renormalize_split_pattern.cc | 2 +-
src/target/llvm/llvm_instance.cc | 4 +-
.../hexagon/hexagon_thread_manager_tests.cc | 4 +-
tests/cpp/parallel_for_test.cc | 3 +-
tests/cpp/te_compute_test.cc | 3 +-
tests/lint/add_asf_header.py | 240 +---
tests/lint/blocklint.sh | 16 +-
.../{add_asf_header.py => check_asf_header.py} | 283 ++++-
tests/lint/check_asf_header.sh | 66 +-
tests/lint/check_cmake_options.py | 1 -
tests/lint/check_file_type.py | 37 +-
tests/lint/filter_untracked.py | 70 --
tests/lint/flake8.sh | 39 +-
tests/lint/git-black.sh | 65 +-
tests/lint/git-clang-format.sh | 79 +-
tests/lint/jnilint.sh | 3 +-
tests/lint/pylint.sh | 38 +-
tests/lint/pylintrc | 428 +-------
tests/lint/rat-excludes | 71 +-
tests/lint/trailing_newlines.py | 35 +-
tests/lint/whitespace.sh | 23 +-
.../test_minimal_target_codegen_llvm.py | 13 +-
.../test_runtime_ndarray.py | 2 +-
.../test_runtime_packed_func.py | 1 +
.../python/arith/test_arith_canonical_simplify.py | 4 +-
tests/python/arith/test_arith_const_int_bound.py | 1 -
tests/python/arith/test_arith_deduce_bound.py | 2 +-
tests/python/arith/test_arith_intset.py | 34 +-
tests/python/arith/test_arith_iter_affine_map.py | 14 +-
.../test_arith_narrow_predicate_expression.py | 2 -
tests/python/arith/test_arith_rewrite_simplify.py | 3 +-
tests/python/arith/test_arith_simplify.py | 2 +-
.../arith/test_arith_solve_linear_equations.py | 6 +-
.../arith/test_arith_solve_linear_inequality.py | 6 +-
tests/python/ci/sample_prs/pr10786.json | 225 ++--
tests/python/ci/test_ci.py | 11 +-
tests/python/ci/test_tvmbot.py | 4 +-
tests/python/ci/test_utils.py | 5 +-
tests/python/codegen/test_gpu_codegen_allreduce.py | 11 +-
tests/python/codegen/test_inject_ptx_ldg32.py | 9 +-
tests/python/codegen/test_target_codegen.py | 3 +-
.../python/codegen/test_target_codegen_aarch64.py | 16 +-
tests/python/codegen/test_target_codegen_arm.py | 6 +-
tests/python/codegen/test_target_codegen_blob.py | 7 +-
tests/python/codegen/test_target_codegen_c_host.py | 26 +-
.../codegen/test_target_codegen_cross_llvm.py | 14 +-
tests/python/codegen/test_target_codegen_cuda.py | 2 +-
.../python/codegen/test_target_codegen_cuda_fp8.py | 9 +-
tests/python/codegen/test_target_codegen_device.py | 6 +-
tests/python/codegen/test_target_codegen_extern.py | 6 +-
.../codegen/test_target_codegen_gpu_common.py | 3 +-
.../python/codegen/test_target_codegen_hexagon.py | 7 +-
tests/python/codegen/test_target_codegen_llvm.py | 18 +-
.../python/codegen/test_target_codegen_llvm_vla.py | 1 +
tests/python/codegen/test_target_codegen_metal.py | 3 +-
tests/python/codegen/test_target_codegen_opencl.py | 5 +-
tests/python/codegen/test_target_codegen_riscv.py | 1 +
tests/python/codegen/test_target_codegen_rocm.py | 6 +-
.../codegen/test_target_codegen_static_init.py | 7 +-
tests/python/codegen/test_target_codegen_vulkan.py | 4 +-
tests/python/codegen/test_target_codegen_x86.py | 3 +-
tests/python/conftest.py | 1 +
tests/python/contrib/test_android/__init__.py | 3 +-
.../python/contrib/test_android/infrastructure.py | 3 +-
.../contrib/test_android/test_meta_schedule.py | 4 +-
tests/python/contrib/test_cblas.py | 10 +-
tests/python/contrib/test_ccache.py | 13 +-
tests/python/contrib/test_coreml_runtime.py | 12 +-
tests/python/contrib/test_dlpack.py | 3 +-
tests/python/contrib/test_edgetpu_runtime.py | 9 +-
tests/python/contrib/test_hexagon/__init__.py | 3 +-
.../python/contrib/test_hexagon/benchmark_util.py | 2 +-
tests/python/contrib/test_hexagon/conftest.py | 4 +-
.../python/contrib/test_hexagon/conv2d/__init__.py | 3 +-
.../python/contrib/test_hexagon/infrastructure.py | 1 +
tests/python/contrib/test_hexagon/pytest_util.py | 5 +-
.../test_hexagon/test_2d_physical_buffers.py | 3 +-
.../test_hexagon/test_async_dma_pipeline.py | 3 +-
.../test_hexagon/test_benchmark_elemwise_add.py | 4 +-
.../test_hexagon/test_benchmark_maxpool2d.py | 12 +-
.../contrib/test_hexagon/test_dma_builtin.py | 26 +-
.../test_hexagon/test_fixed_point_conversion.py | 2 +
.../contrib/test_hexagon/test_memory_alloc.py | 7 +-
.../contrib/test_hexagon/test_meta_schedule.py | 10 +-
.../contrib/test_hexagon/test_parallel_hvx.py | 1 +
.../test_hexagon/test_parallel_hvx_load_vtcm.py | 1 +
.../contrib/test_hexagon/test_parallel_scalar.py | 6 +-
.../contrib/test_hexagon/test_relax_integration.py | 3 +-
.../contrib/test_hexagon/test_run_unit_tests.py | 8 +-
tests/python/contrib/test_hexagon/test_sigmoid.py | 11 +-
tests/python/contrib/test_hexagon/test_take.py | 7 +-
.../contrib/test_hexagon/test_thread_pool.py | 2 +-
tests/python/contrib/test_hexagon/test_vtcm.py | 14 +-
.../contrib/test_hexagon/test_vtcm_bandwidth.py | 2 +-
tests/python/contrib/test_memoize.py | 34 +-
tests/python/contrib/test_mps.py | 3 +-
tests/python/contrib/test_msc/test_graph_build.py | 8 +-
tests/python/contrib/test_msc/test_pipeline.py | 11 +-
tests/python/contrib/test_msc/test_plugin.py | 21 +-
tests/python/contrib/test_msc/test_runner.py | 13 +-
tests/python/contrib/test_msc/test_tools.py | 11 +-
tests/python/contrib/test_msc/test_transform.py | 13 +-
.../contrib/test_msc/test_translate_relax.py | 10 +-
.../contrib/test_msc/test_translate_tensorrt.py | 9 +-
.../contrib/test_msc/test_translate_torch.py | 10 +-
tests/python/contrib/test_popen_pool.py | 19 +-
tests/python/contrib/test_random.py | 8 +-
tests/python/contrib/test_rocblas.py | 4 +-
tests/python/contrib/test_rpc_proxy.py | 4 +-
tests/python/contrib/test_rpc_tracker.py | 4 +-
tests/python/contrib/test_sort.py | 2 +
tests/python/contrib/test_tflite_runtime.py | 6 +-
tests/python/contrib/test_util.py | 1 +
tests/python/disco/test_callback.py | 4 +-
tests/python/disco/test_ccl.py | 32 +-
tests/python/disco/test_loader.py | 12 +-
tests/python/disco/test_nvshmem.py | 37 +-
tests/python/disco/test_session.py | 21 +-
tests/python/ir/analysis/test_collect_call_map.py | 6 +-
tests/python/ir/test_container_structural_equal.py | 6 +-
tests/python/ir/test_dtype.py | 1 +
tests/python/ir/test_ir_attrs.py | 3 +-
tests/python/ir/test_ir_container.py | 4 +-
tests/python/ir/test_ir_type.py | 1 +
tests/python/ir/test_node_reflection.py | 8 +-
tests/python/ir/test_pass_instrument.py | 3 +-
tests/python/ir/test_roundtrip_runtime_module.py | 3 +-
.../python/ir/test_transform_replace_global_var.py | 4 +-
tests/python/nightly/test_nnapi/infrastructure.py | 5 +-
.../test_nnapi/test_from_exported_to_cuda.py | 15 +-
tests/python/nightly/test_nnapi/test_network.py | 4 +-
tests/python/nightly/test_nnapi/test_ops.py | 4 +-
.../adreno/test_transform_annotate_custom_scope.py | 44 +-
.../test_transform_fold_vdevice_scope_change.py | 20 +-
tests/python/relax/backend/clml/conftest.py | 4 +-
tests/python/relax/backend/clml/mod_utils.py | 12 +-
.../python/relax/backend/clml/test_clml_codegen.py | 49 +-
.../backend/clml/test_op_exec_clml_codegen.py | 37 +-
tests/python/relax/backend/clml/utils.py | 19 +-
.../distributed/test_distributed_dtensor_sinfo.py | 9 +-
..._distributed_transform_legalize_redistribute.py | 12 +-
.../test_distributed_transform_lower_distir.py | 8 +-
...ributed_transform_lower_global_to_local_view.py | 386 +++----
...est_distributed_transform_propagate_sharding.py | 8 +-
.../test_distributed_tvmscript_parser.py | 5 +-
.../test_distributed_tvmscript_printer.py | 4 +-
.../test_runtime_builtin_kv_cache_transfer.py | 7 +-
...est_runtime_builtin_kv_cache_transfer_kernel.py | 4 +-
tests/python/relax/test_analysis.py | 3 +-
.../test_analysis_computable_at_compile_time.py | 3 +-
.../python/relax/test_analysis_detect_recursion.py | 5 +-
.../relax/test_analysis_estimate_memory_usage.py | 3 +-
.../relax/test_analysis_struct_info_analysis.py | 12 +-
.../test_analysis_suggest_layout_transforms.py | 4 +-
tests/python/relax/test_analysis_well_formed.py | 5 +-
tests/python/relax/test_ast_printer.py | 3 +-
.../relax/test_backend_dispatch_sort_scan.py | 2 +-
.../relax/test_backend_transform_shape_lower.py | 42 +-
tests/python/relax/test_base_py_module.py | 6 +-
tests/python/relax/test_base_py_module_printer.py | 18 +-
.../relax/test_base_py_module_symbolic_shape.py | 12 +-
tests/python/relax/test_bind_params.py | 6 +-
tests/python/relax/test_bind_symbolic_vars.py | 9 +-
tests/python/relax/test_binding_rewrite.py | 1 +
tests/python/relax/test_blockbuilder_core.py | 11 +-
tests/python/relax/test_blockbuilder_emit_te.py | 11 +-
tests/python/relax/test_codegen_cudnn.py | 4 +-
tests/python/relax/test_codegen_cutlass.py | 6 +-
tests/python/relax/test_codegen_dnnl.py | 8 +-
tests/python/relax/test_codegen_tensorrt.py | 8 +-
tests/python/relax/test_contrib_vllm.py | 3 +-
tests/python/relax/test_dataflow_inplace.py | 19 +-
tests/python/relax/test_dataflow_rewriter.py | 8 +-
tests/python/relax/test_dlpack_integration.py | 7 +-
tests/python/relax/test_e2e_op_dynamic.py | 8 +-
...eliminate_pad_branch_using_buffer_assumption.py | 132 ++-
tests/python/relax/test_expr.py | 3 +-
tests/python/relax/test_expr_args_converter.py | 5 +-
tests/python/relax/test_expr_functor.py | 23 +-
tests/python/relax/test_fast_math_transform.py | 4 +-
tests/python/relax/test_frontend_common.py | 18 +-
tests/python/relax/test_frontend_dynamo.py | 19 +-
.../relax/test_frontend_from_exported_program.py | 863 +++++++--------
tests/python/relax/test_frontend_from_fx.py | 500 +++++----
tests/python/relax/test_frontend_nn_exporter.py | 37 +-
tests/python/relax/test_frontend_nn_jit.py | 5 +-
tests/python/relax/test_frontend_nn_modules.py | 36 +-
tests/python/relax/test_frontend_nn_op.py | 17 +-
tests/python/relax/test_frontend_onnx.py | 14 +-
tests/python/relax/test_frontend_stablehlo.py | 11 +-
tests/python/relax/test_group_gemm_flashinfer.py | 12 +-
tests/python/relax/test_inline_functions.py | 4 +-
tests/python/relax/test_kill_after_last_use.py | 5 +-
tests/python/relax/test_lower_alloc_tensor.py | 5 +-
.../relax/test_meta_schedule_relax_integration.py | 6 +-
tests/python/relax/test_op_binary.py | 5 +-
tests/python/relax/test_op_ccl.py | 4 +-
tests/python/relax/test_op_datatype.py | 6 +-
tests/python/relax/test_op_distributed.py | 3 +-
tests/python/relax/test_op_grad.py | 3 +-
tests/python/relax/test_op_gradient_numeric.py | 11 +-
tests/python/relax/test_op_image.py | 4 +-
tests/python/relax/test_op_index.py | 10 +-
tests/python/relax/test_op_linear_algebra.py | 4 +-
tests/python/relax/test_op_manipulate.py | 9 +-
tests/python/relax/test_op_nn_convolution.py | 4 +-
tests/python/relax/test_op_nn_pooling.py | 4 +-
tests/python/relax/test_op_search.py | 4 +-
tests/python/relax/test_op_set.py | 4 +-
tests/python/relax/test_op_sort.py | 4 +-
tests/python/relax/test_op_statistical.py | 5 +-
tests/python/relax/test_op_take.py | 8 +-
tests/python/relax/test_op_ternary.py | 4 +-
tests/python/relax/test_op_unary.py | 5 +-
tests/python/relax/test_op_view.py | 10 +-
.../python/relax/test_optimize_layout_transform.py | 5 +-
tests/python/relax/test_pytorch_integration.py | 7 +-
tests/python/relax/test_relax_operators.py | 4 +-
.../python/relax/test_relax_to_pyfunc_converter.py | 135 ++-
.../python/relax/test_remove_redundant_reshape.py | 43 +-
tests/python/relax/test_runtime_builtin.py | 9 +-
...runtime_builtin_paged_attention_kv_cache_cpu.py | 7 +-
..._builtin_paged_attention_kv_cache_flashinfer.py | 2 +-
...ltin_paged_attention_kv_cache_mla_flashinfer.py | 2 +-
...ime_builtin_paged_attention_kv_cache_mla_tir.py | 2 +-
...runtime_builtin_paged_attention_kv_cache_tir.py | 7 +-
.../python/relax/test_runtime_builtin_rnn_state.py | 2 +-
.../relax/test_runtime_sampling_flashinfer.py | 6 +-
tests/python/relax/test_struct_info.py | 7 +-
tests/python/relax/test_testing_nn.py | 10 +-
tests/python/relax/test_training_append_loss.py | 4 +-
tests/python/relax/test_training_loss.py | 3 +-
tests/python/relax/test_training_optimizer.py | 4 +-
.../relax/test_training_optimizer_numeric.py | 1 +
tests/python/relax/test_training_setup_trainer.py | 8 +-
.../python/relax/test_training_trainer_numeric.py | 11 +-
tests/python/relax/test_transform.py | 8 +-
.../relax/test_transform_adjust_matmul_order.py | 4 +-
tests/python/relax/test_transform_alter_op_impl.py | 20 +-
.../test_transform_annotate_tir_op_pattern.py | 1 +
...st_transform_attach_attr_layout_free_buffers.py | 6 +-
.../relax/test_transform_attach_global_symbol.py | 10 +-
tests/python/relax/test_transform_bind_params.py | 7 +-
.../relax/test_transform_bind_symbolic_vars.py | 6 +-
.../relax/test_transform_bundle_model_params.py | 6 +-
.../relax/test_transform_canonicalize_bindings.py | 9 +-
tests/python/relax/test_transform_codegen_pass.py | 6 +-
.../test_transform_combine_parallel_matmul.py | 4 +-
.../relax/test_transform_compute_prim_value.py | 4 +-
.../python/relax/test_transform_convert_layout.py | 64 +-
tests/python/relax/test_transform_cse.py | 9 +-
.../relax/test_transform_dead_code_elimination.py | 52 +-
tests/python/relax/test_transform_decompose_ops.py | 4 +-
.../relax/test_transform_expand_matmul_of_sum.py | 3 +-
.../relax/test_transform_expand_tuple_args.py | 6 +-
.../python/relax/test_transform_few_shot_tuning.py | 13 +-
.../test_transform_fold_batch_norm_to_conv2d.py | 12 +-
tests/python/relax/test_transform_fold_constant.py | 36 +-
tests/python/relax/test_transform_fuse_ops.py | 10 +-
.../relax/test_transform_fuse_ops_by_pattern.py | 44 +-
tests/python/relax/test_transform_fuse_tir.py | 36 +-
.../relax/test_transform_fuse_transpose_matmul.py | 3 +-
tests/python/relax/test_transform_gradient.py | 6 +-
.../relax/test_transform_gradient_checkpoint.py | 10 +-
.../relax/test_transform_gradient_numeric.py | 4 +-
.../relax/test_transform_gradient_te_register.py | 6 +-
tests/python/relax/test_transform_lambda_lift.py | 80 +-
.../relax/test_transform_lazy_transform_params.py | 30 +-
tests/python/relax/test_transform_legalize_ops.py | 8 +-
.../relax/test_transform_legalize_ops_binary.py | 7 +-
.../test_transform_legalize_ops_create_datatype.py | 6 +-
.../relax/test_transform_legalize_ops_grad.py | 8 +-
.../relax/test_transform_legalize_ops_image.py | 5 +-
.../test_transform_legalize_ops_manipulate.py | 11 +-
.../relax/test_transform_legalize_ops_qdq.py | 5 +-
...st_transform_legalize_ops_search_statistical.py | 8 +-
.../relax/test_transform_legalize_ops_unary.py | 5 +-
.../relax/test_transform_lift_transform_params.py | 13 +-
.../test_transform_merge_composite_functions.py | 50 +-
.../test_transform_meta_schedule_apply_database.py | 6 +-
.../relax/test_transform_meta_schedule_tuning.py | 3 +-
tests/python/relax/test_transform_normalize.py | 33 +-
.../relax/test_transform_normalize_global_var.py | 10 +-
...st_transform_operator_specific_normalization.py | 23 +-
.../python/relax/test_transform_realize_vdevice.py | 4 +-
.../relax/test_transform_remove_unused_outputs.py | 10 +-
.../test_transform_remove_unused_parameters.py | 10 +-
..._transform_reorder_permute_dims_after_concat.py | 3 +-
.../test_transform_reorder_take_after_matmul.py | 4 +-
.../relax/test_transform_rewrite_cuda_graph.py | 8 +-
.../test_transform_rewrite_dataflow_reshape.py | 28 +-
...nsform_specialize_primfunc_based_on_callsite.py | 76 +-
.../test_transform_static_plan_block_memory.py | 12 +-
.../relax/test_transform_to_mixed_precision.py | 7 +-
.../relax/test_transform_topological_sort.py | 3 +-
.../test_transform_update_param_struct_info.py | 3 +-
.../python/relax/test_transform_update_vdevice.py | 28 +-
tests/python/relax/test_tvmscript_ir_builder.py | 6 +-
tests/python/relax/test_tvmscript_parser.py | 45 +-
.../relax/test_tvmscript_parser_op_arith_cmp.py | 14 +-
.../relax/test_tvmscript_parser_op_create.py | 6 +-
.../python/relax/test_tvmscript_parser_op_index.py | 6 +-
.../relax/test_tvmscript_parser_op_manipulate.py | 17 +-
tests/python/relax/test_tvmscript_parser_op_nn.py | 30 +-
tests/python/relax/test_tvmscript_parser_op_set.py | 4 +-
.../python/relax/test_tvmscript_parser_op_sort.py | 6 +-
.../relax/test_tvmscript_parser_op_statistical.py | 6 +-
.../relax/test_tvmscript_parser_op_vision.py | 18 +-
tests/python/relax/test_tvmscript_pyfunc.py | 7 +-
tests/python/relax/test_utils.py | 3 +-
tests/python/relax/test_vm_build.py | 18 +-
tests/python/relax/test_vm_builtin.py | 3 +-
tests/python/relax/test_vm_builtin_lower.py | 7 +-
tests/python/relax/test_vm_callback_function.py | 11 +-
tests/python/relax/test_vm_codegen_only.py | 2 +
tests/python/relax/test_vm_codegen_tir.py | 1 +
tests/python/relax/test_vm_cuda_graph.py | 12 +-
tests/python/relax/test_vm_execbuilder.py | 1 +
tests/python/relax/test_vm_instrument.py | 2 +-
tests/python/relax/test_vm_multi_device.py | 27 +-
tests/python/relax/test_vm_profiler.py | 6 +-
tests/python/relax/texture/adreno_utils.py | 15 +-
tests/python/relax/texture/test_network.py | 22 +-
tests/python/relax/texture/test_ops.py | 7 +-
tests/python/relax/texture/test_texture_nd.py | 19 +-
.../python/runtime/test_evaluator_with_preproc.py | 2 +-
tests/python/runtime/test_rpc_base.py | 6 +-
tests/python/runtime/test_runtime_container.py | 2 +-
tests/python/runtime/test_runtime_dlpack.py | 3 +-
tests/python/runtime/test_runtime_error.py | 2 +-
tests/python/runtime/test_runtime_extension.py | 6 +-
tests/python/runtime/test_runtime_measure.py | 6 +-
tests/python/runtime/test_runtime_module_export.py | 1 -
tests/python/runtime/test_runtime_module_load.py | 15 +-
.../python/runtime/test_runtime_module_property.py | 2 +-
tests/python/runtime/test_runtime_nd_array.py | 6 +-
tests/python/runtime/test_runtime_rpc.py | 14 +-
tests/python/runtime/test_runtime_trace.py | 25 +-
...st_s_tir_analysis_calculate_allocated_memory.py | 12 +-
.../test_s_tir_analysis_estimate_tir_flops.py | 2 +-
.../test_s_tir_analysis_identify_memcpy.py | 4 +-
.../test_s_tir_analysis_is_pure_function.py | 3 +-
.../s_tir/base/test_sblock_dependence_info.py | 17 +-
tests/python/s_tir/base/test_tir_data_layout.py | 1 +
.../s_tir/base/test_tir_te_extern_primfunc.py | 4 +-
tests/python/s_tir/dlight/test_benchmark.py | 16 +-
tests/python/s_tir/dlight/test_gpu_fallback.py | 2 +-
.../s_tir/dlight/test_gpu_general_reduction.py | 14 +-
.../python/s_tir/dlight/test_gpu_low_batch_gemv.py | 1 +
.../s_tir/dlight/test_gpu_matmul_tensorize.py | 1 -
tests/python/s_tir/dlight/test_gpu_reduction.py | 2 +-
tests/python/s_tir/dlight/test_gpu_rmsnorm.py | 3 +-
tests/python/s_tir/dlight/test_gpu_transpose.py | 2 +-
.../meta_schedule/test_meta_schedule_builder.py | 7 +-
.../meta_schedule/test_meta_schedule_cost_model.py | 3 +-
.../meta_schedule/test_meta_schedule_database.py | 6 +-
.../test_meta_schedule_feature_extractor.py | 1 +
...schedule_feature_extractor_per_store_feature.py | 24 +-
.../test_meta_schedule_measure_callback.py | 3 +-
.../test_meta_schedule_mma_tensorize.py | 10 +-
...eta_schedule_mutator_mutate_compute_location.py | 2 +-
.../test_meta_schedule_mutator_mutate_parallel.py | 2 +-
..._meta_schedule_mutator_mutate_thread_binding.py | 2 +-
.../test_meta_schedule_mutator_mutate_tile_size.py | 2 +-
.../test_meta_schedule_mutator_mutate_unroll.py | 2 +-
.../test_meta_schedule_post_order_apply.py | 7 +-
...ule_postproc_disallow_async_strided_mem_copy.py | 2 +-
...meta_schedule_postproc_disallow_dynamic_loop.py | 2 +-
..._schedule_postproc_rewrite_cooperative_fetch.py | 2 +-
.../test_meta_schedule_postproc_rewrite_layout.py | 3 +-
...e_postproc_rewrite_parallel_vectorize_unroll.py | 2 +-
...ta_schedule_postproc_rewrite_reduction_block.py | 2 +-
...est_meta_schedule_postproc_rewrite_tensorize.py | 2 +-
...meta_schedule_postproc_rewrite_unbound_block.py | 2 +-
.../test_meta_schedule_postproc_verify_gpu_code.py | 3 +-
...est_meta_schedule_postproc_verify_vtcm_limit.py | 2 +-
.../meta_schedule/test_meta_schedule_profiler.py | 3 +-
.../meta_schedule/test_meta_schedule_runner.py | 7 +-
...eta_schedule_schedule_rule_apply_custom_rule.py | 3 +-
...test_meta_schedule_schedule_rule_auto_inline.py | 4 +-
.../test_meta_schedule_schedule_rule_mlt.py | 2 +-
.../test_meta_schedule_schedule_rule_mlt_intrin.py | 6 +-
.../test_meta_schedule_schedule_rule_mlt_tc.py | 7 +-
.../test_meta_schedule_search_strategy.py | 9 +-
.../meta_schedule/test_meta_schedule_space_cpu.py | 3 +-
.../meta_schedule/test_meta_schedule_space_cuda.py | 1 +
.../test_meta_schedule_space_cuda_async.py | 1 +
...ule_space_cuda_async_multiple_initialization.py | 7 +-
.../test_meta_schedule_space_generator.py | 3 +-
.../test_meta_schedule_task_scheduler.py | 3 +-
.../test_meta_schedule_trace_apply.py | 7 +-
.../test_meta_schedule_tune_context.py | 3 +-
.../meta_schedule/test_meta_schedule_tune_tir.py | 2 +-
.../s_tir/schedule/test_tir_schedule_analysis.py | 27 +-
.../test_tir_schedule_annotate_buffer_access.py | 4 +-
.../schedule/test_tir_schedule_block_scope.py | 5 +-
.../s_tir/schedule/test_tir_schedule_blockize.py | 5 +-
.../schedule/test_tir_schedule_cache_index.py | 3 +-
.../schedule/test_tir_schedule_cache_read_write.py | 43 +-
.../s_tir/schedule/test_tir_schedule_compute_at.py | 5 +-
.../schedule/test_tir_schedule_compute_inline.py | 13 +-
.../test_tir_schedule_decompose_padding.py | 1 +
.../s_tir/schedule/test_tir_schedule_error.py | 2 +-
.../s_tir/schedule/test_tir_schedule_for_kind.py | 5 +-
.../test_tir_schedule_fuse_reduction_epilogue.py | 7 +-
...ir_schedule_fuse_reduction_epilogue_clipping.py | 7 +-
...st_tir_schedule_fuse_reduction_epilogue_relu.py | 7 +-
.../schedule/test_tir_schedule_instruction.py | 3 +-
.../s_tir/schedule/test_tir_schedule_merge.py | 5 +-
.../s_tir/schedule/test_tir_schedule_pad_einsum.py | 5 +-
.../s_tir/schedule/test_tir_schedule_partition.py | 5 +-
.../schedule/test_tir_schedule_read_write_at.py | 22 +-
.../s_tir/schedule/test_tir_schedule_reduction.py | 9 +-
.../s_tir/schedule/test_tir_schedule_reindex.py | 3 +-
.../s_tir/schedule/test_tir_schedule_reorder.py | 3 +-
.../test_tir_schedule_reorder_block_iter_var.py | 3 +-
.../s_tir/schedule/test_tir_schedule_rfactor.py | 26 +-
.../schedule/test_tir_schedule_rolling_buffer.py | 7 +-
.../s_tir/schedule/test_tir_schedule_sampling.py | 7 +-
.../test_tir_schedule_set_axis_separator.py | 7 +-
.../s_tir/schedule/test_tir_schedule_set_dtype.py | 3 +-
.../s_tir/schedule/test_tir_schedule_set_scope.py | 3 +-
.../s_tir/schedule/test_tir_schedule_split_fuse.py | 7 +-
.../s_tir/schedule/test_tir_schedule_state.py | 1 +
.../test_tir_schedule_state_cached_flags.py | 5 +-
.../schedule/test_tir_schedule_storage_align.py | 3 +-
.../s_tir/schedule/test_tir_schedule_tensorize.py | 11 +-
..._tir_schedule_tensorize_ldmatrix_mma_numeric.py | 2 +-
.../test_tir_schedule_tensorize_mfma_numeric.py | 27 +-
.../s_tir/schedule/test_tir_schedule_trace.py | 7 +-
.../s_tir/schedule/test_tir_schedule_transform.py | 4 +-
.../schedule/test_tir_schedule_transform_layout.py | 5 +-
.../s_tir/schedule/test_tir_schedule_utilities.py | 3 +-
...test_s_tir_transform_annotate_irregular_loop.py | 5 +-
.../test_s_tir_transform_canonicalize_loop.py | 3 +-
.../test_s_tir_transform_compact_buffer_region.py | 2 +-
...est_s_tir_transform_convert_blocks_to_opaque.py | 5 +-
.../test_s_tir_transform_default_gpu_schedule.py | 2 +-
.../test_s_tir_transform_hoist_expression.py | 2 +-
.../transform/test_s_tir_transform_hoist_if.py | 8 +-
.../test_s_tir_transform_inject_double_buffer.py | 4 +-
.../test_s_tir_transform_inject_permuted_layout.py | 2 +-
.../test_s_tir_transform_inject_ptx_async_copy.py | 17 +-
...est_s_tir_transform_inject_software_pipeline.py | 7 +-
.../test_s_tir_transform_inject_virtual_thread.py | 3 +-
.../test_s_tir_transform_lift_thread_binding.py | 2 +-
.../test_s_tir_transform_loop_partition.py | 3 +-
...s_tir_transform_lower_cross_thread_reduction.py | 13 +-
.../test_s_tir_transform_lower_match_buffer.py | 2 +-
.../test_s_tir_transform_lower_opaque_block.py | 2 +-
...test_s_tir_transform_lower_thread_all_reduce.py | 3 +-
...transform_manifest_shared_memory_local_stage.py | 1 -
...st_s_tir_transform_memhammer_lower_auto_copy.py | 14 +-
...form_merge_dynamic_shared_memory_allocations.py | 27 +-
...sform_plan_update_buffer_allocation_location.py | 4 +-
.../test_s_tir_transform_profiling_instr.py | 3 +-
.../transform/test_s_tir_transform_remove_undef.py | 4 +-
.../test_s_tir_transform_rewrite_unsafe_select.py | 3 +-
.../test_s_tir_transform_unify_thread_binding.py | 3 +-
tests/python/target/test_arm_target.py | 4 +-
tests/python/target/test_llvm_features_info.py | 2 +-
tests/python/target/test_riscv_features.py | 4 +-
tests/python/target/test_target_target.py | 1 +
tests/python/target/test_virtual_device.py | 1 +
tests/python/target/test_x86_features.py | 2 +-
tests/python/te/test_te_create_primfunc.py | 5 +-
tests/python/te/test_te_tag.py | 2 +-
tests/python/te/test_te_tensor.py | 1 +
tests/python/testing/test_filter_untracked.py | 196 ----
tests/python/testing/test_testing.py | 1 +
.../testing/test_tvm_testing_before_after.py | 3 +-
.../python/testing/test_type_annotation_checker.py | 10 +-
.../test_tir_analysis_verify_well_formed.py | 3 +-
tests/python/tir-base/test_tir_base.py | 7 +-
tests/python/tir-base/test_tir_buffer.py | 8 +-
tests/python/tir-base/test_tir_constructor.py | 1 +
tests/python/tir-base/test_tir_host_func.py | 8 +-
tests/python/tir-base/test_tir_imm_values.py | 5 +-
tests/python/tir-base/test_tir_index_map.py | 2 +-
tests/python/tir-base/test_tir_intrin.py | 15 +-
tests/python/tir-base/test_tir_nodes.py | 1 +
tests/python/tir-base/test_tir_ops.py | 6 +-
tests/python/tir-base/test_tir_ptx_cp_async.py | 7 +-
tests/python/tir-base/test_tir_ptx_ldmatrix.py | 5 +-
tests/python/tir-base/test_tir_ptx_mma.py | 6 +-
tests/python/tir-base/test_tir_ptx_mma_sp.py | 5 +-
tests/python/tir-base/test_tir_specialize.py | 2 +-
.../tir-base/test_tir_stmt_functor_ir_transform.py | 3 +-
.../tir-base/test_tir_stmt_functor_substitute.py | 4 +-
.../tir-base/test_tir_structural_equal_hash.py | 19 +-
tests/python/tir-base/test_tir_texture_scope.py | 2 +-
.../tir-base/test_tir_unsafe_hide_buffer_access.py | 3 +-
.../test_tir_inline_private_functions.py | 3 +-
.../test_tir_transform_annotate_device_regions.py | 3 +-
.../test_tir_transform_bf16_legalize.py | 1 -
.../test_tir_transform_convert_ssa.py | 5 +-
.../test_tir_transform_device_kernel_launch.py | 3 +-
.../test_tir_transform_flatten_buffer.py | 3 +-
...test_tir_transform_force_narrow_index_to_i32.py | 3 +-
.../tir-transform/test_tir_transform_helpers.py | 3 +-
.../test_tir_transform_lower_intrin.py | 3 +-
.../test_tir_transform_lower_tvm_builtin.py | 7 +-
.../test_tir_transform_make_packed_api.py | 6 +-
.../test_tir_transform_narrow_datatype.py | 8 +-
...est_tir_transform_pointer_value_type_rewrite.py | 3 +-
.../test_tir_transform_remove_assume.py | 3 +-
.../test_tir_transform_remove_no_op.py | 6 +-
.../tir-transform/test_tir_transform_simplify.py | 3 +-
.../test_tir_transform_storage_rewrite.py | 3 +-
.../test_tir_transform_unroll_loop.py | 3 +-
.../tir-transform/test_tir_transform_vectorize.py | 10 +-
.../tvmscript/test_tvmscript_error_report.py | 17 +-
.../tvmscript/test_tvmscript_ir_builder_base.py | 2 +
.../test_tvmscript_ir_builder_irmodule.py | 6 +-
.../tvmscript/test_tvmscript_ir_builder_tir.py | 9 +-
tests/python/tvmscript/test_tvmscript_ops.py | 1 +
.../tvmscript/test_tvmscript_parser_evaluator.py | 2 +
tests/python/tvmscript/test_tvmscript_parser_ir.py | 6 +-
.../tvmscript/test_tvmscript_parser_source.py | 9 +-
.../python/tvmscript/test_tvmscript_parser_tir.py | 5 +-
.../tvmscript/test_tvmscript_printer_annotation.py | 1 +
.../python/tvmscript/test_tvmscript_printer_doc.py | 3 +-
.../tvmscript/test_tvmscript_printer_highlight.py | 2 +-
.../test_tvmscript_printer_python_doc_printer.py | 1 +
.../test_tvmscript_printer_structural_equal.py | 5 +-
.../python/tvmscript/test_tvmscript_printer_tir.py | 34 +-
.../test_tvmscript_printer_underlining.py | 4 +-
.../python/tvmscript/test_tvmscript_regression.py | 1 +
tests/python/tvmscript/test_tvmscript_roundtrip.py | 1146 ++++++++++----------
.../tvmscript/test_tvmscript_syntax_sugar.py | 3 +-
tests/scripts/ci.py | 20 +-
tests/scripts/release/gather_prs.py | 11 +-
tests/scripts/release/make_notes.py | 8 +-
tests/scripts/request_hook/request_hook.py | 3 +-
tests/scripts/task_build.py | 11 +-
tests/scripts/task_lint.sh | 48 +-
tests/scripts/task_mypy.sh | 26 +-
version.py | 5 +-
web/tests/python/prepare_test_libs.py | 4 +-
web/tests/python/relax_rpc_test.py | 7 +-
web/tests/python/webgpu_rpc_test.py | 8 +-
1450 files changed, 8322 insertions(+), 8044 deletions(-)
copy .github/workflows/{update_nightly_branch.yml => lint.yml} (64%)
copy python/tvm/contrib/msc/__init__.py => .markdownlint-cli2.yaml (97%)
copy docker/install/ubuntu_install_cuda_python.sh => .yamllint.yaml (78%)
create mode 100644 docs/.rstcheck.cfg
delete mode 100644 mypy.ini
copy tests/lint/{add_asf_header.py => check_asf_header.py} (53%)
delete mode 100644 tests/lint/filter_untracked.py
mode change 100755 => 100644 tests/lint/trailing_newlines.py
delete mode 100644 tests/python/testing/test_filter_untracked.py
