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 0079ff7d1c [REFACTOR][TEST] Replace CompareBeforeAfter for pytest
compact (#18711)
add d76c729259 [REFACTOR][S-TIR] Initialize the s_tir module (#18712)
add dcad7373c7 [Relax][Torch] Avoid decomposition crash with sparse CSR
buffers (#18670)
add 739027de42 [SPIRV] Fix forloop codegen in vulkan (#18715)
No new revisions were added by this update.
Summary of changes:
apps/android_rpc/tests/android_rpc_test.py | 2 +-
apps/ios_rpc/tests/ios_rpc_test.py | 2 +-
.../tensor_ir/tutorials/tir_transformation.py | 2 +-
docs/how_to/tutorials/cross_compilation_and_rpc.py | 2 +-
docs/reference/api/python/index.rst | 7 +-
.../api/python/{tir => s_tir}/schedule.rst | 6 +-
jvm/core/src/test/scripts/prepare_test_libs.py | 2 +-
mypy.ini | 2 +-
python/tvm/__init__.py | 3 +
python/tvm/dlight/adreno/convolution.py | 14 +-
python/tvm/dlight/adreno/fallback.py | 12 +-
python/tvm/dlight/adreno/layout_transform.py | 12 +-
python/tvm/dlight/adreno/pool.py | 10 +-
python/tvm/dlight/adreno/utils.py | 6 +-
python/tvm/dlight/analysis/common_analysis.py | 46 +++---
python/tvm/dlight/analysis/gemv.py | 8 +-
python/tvm/dlight/base/common_schedules.py | 10 +-
python/tvm/dlight/base/schedule_rule.py | 10 +-
python/tvm/dlight/base/transform.py | 6 +-
python/tvm/dlight/base/utils.py | 6 +-
python/tvm/dlight/cpu/gemv.py | 12 +-
python/tvm/dlight/gpu/fallback.py | 14 +-
python/tvm/dlight/gpu/gemv.py | 22 +--
python/tvm/dlight/gpu/general_reduction.py | 6 +-
python/tvm/dlight/gpu/low_batch_gemv.py | 34 ++---
python/tvm/dlight/gpu/matmul.py | 56 ++++----
python/tvm/dlight/gpu/reduction.py | 16 +--
python/tvm/dlight/gpu/rmsnorm.py | 4 +-
python/tvm/dlight/gpu/transpose.py | 10 +-
python/tvm/exec/gpu_memory_bandwidth.py | 6 +-
python/tvm/meta_schedule/builder/local_builder.py | 2 +-
python/tvm/meta_schedule/database/database.py | 8 +-
.../meta_schedule/database/schedule_fn_database.py | 2 +-
python/tvm/meta_schedule/mutator/mutator.py | 2 +-
.../tvm/meta_schedule/post_optimization/space.py | 4 +-
python/tvm/meta_schedule/postproc/postproc.py | 4 +-
.../schedule/cuda/layout_transform.py | 12 +-
.../schedule_rule/multi_level_tiling.py | 2 +-
.../meta_schedule/schedule_rule/schedule_rule.py | 6 +-
.../search_strategy/search_strategy.py | 8 +-
.../space_generator/space_generator.py | 6 +-
python/tvm/meta_schedule/testing/dummy_object.py | 2 +-
.../tvm/meta_schedule/testing/space_generation.py | 6 +-
python/tvm/meta_schedule/testing/tune_te.py | 3 +-
.../tvm/meta_schedule/testing/validate_database.py | 6 +-
python/tvm/meta_schedule/tir_integration.py | 7 +-
python/tvm/meta_schedule/trace_apply.py | 2 +-
python/tvm/meta_schedule/tune_context.py | 9 +-
python/tvm/relax/frontend/nn/llm/kv_cache.py | 20 +--
python/tvm/relax/frontend/nn/llm/tree_attn.py | 18 +--
.../frontend/torch/exported_program_translator.py | 28 +++-
.../tvm/relax/transform/legalize_ops/manipulate.py | 6 +-
.../conftest.py => python/tvm/s_tir/__init__.py | 15 +-
python/tvm/{tir => s_tir}/_ffi_api.py | 0
python/tvm/{tir => s_tir}/backend/__init__.py | 2 +-
.../hexagon => s_tir/backend/adreno}/__init__.py | 4 +-
.../tvm/{tir => s_tir}/backend/adreno/pipeline.py | 31 +---
python/tvm/{tir => s_tir}/block_dependence_info.py | 0
python/tvm/{tir => s_tir}/block_scope.py | 0
.../tvm/{tir/backend/adreno => s_tir}/pipeline.py | 25 +---
python/tvm/{tir => s_tir}/schedule/__init__.py | 0
python/tvm/{tir => s_tir}/schedule/_ffi_api.py | 2 +-
.../tvm/{tir => s_tir}/schedule/_type_checker.py | 0
python/tvm/{tir => s_tir}/schedule/analysis.py | 8 +-
python/tvm/{tir => s_tir}/schedule/instruction.py | 0
python/tvm/{tir => s_tir}/schedule/schedule.py | 78 +++++-----
python/tvm/{tir => s_tir}/schedule/state.py | 0
python/tvm/{tir => s_tir}/schedule/testing.py | 4 +-
python/tvm/{tir => s_tir}/schedule/trace.py | 4 +-
python/tvm/{tir => s_tir}/schedule/transform.py | 2 +-
.../tvm/{tir => s_tir}/tensor_intrin/__init__.py | 0
python/tvm/{tir => s_tir}/tensor_intrin/arm_cpu.py | 0
python/tvm/{tir => s_tir}/tensor_intrin/cuda.py | 0
.../tensor_intrin/dot_product_common.py | 0
python/tvm/{tir => s_tir}/tensor_intrin/hexagon.py | 0
python/tvm/{tir => s_tir}/tensor_intrin/metal.py | 0
.../tvm/{tir => s_tir}/tensor_intrin/riscv_cpu.py | 0
python/tvm/{tir => s_tir}/tensor_intrin/rocm.py | 0
python/tvm/{tir => s_tir}/tensor_intrin/x86.py | 0
python/tvm/testing/tir.py | 4 +-
python/tvm/tir/__init__.py | 6 +-
python/tvm/tir/backend/__init__.py | 2 +-
python/tvm/tir/backend/adreno/__init__.py | 3 +-
python/tvm/tir/pipeline.py | 120 +---------------
src/target/spirv/codegen_spirv.cc | 8 +-
src/tir/schedule/trace.cc | 4 +-
.../test_minimal_target_codegen_llvm.py | 2 +-
tests/python/codegen/test_gpu_codegen_allreduce.py | 6 +-
tests/python/codegen/test_target_codegen_arm.py | 6 +-
tests/python/codegen/test_target_codegen_bool.py | 2 +-
.../codegen/test_target_codegen_cross_llvm.py | 2 +-
tests/python/codegen/test_target_codegen_cuda.py | 32 ++---
.../python/codegen/test_target_codegen_cuda_fp4.py | 4 +-
.../python/codegen/test_target_codegen_cuda_fp8.py | 12 +-
tests/python/codegen/test_target_codegen_device.py | 4 +-
.../codegen/test_target_codegen_gpu_common.py | 2 +-
tests/python/codegen/test_target_codegen_llvm.py | 40 +++---
tests/python/codegen/test_target_codegen_metal.py | 4 +-
tests/python/codegen/test_target_codegen_opencl.py | 14 +-
tests/python/codegen/test_target_codegen_rocm.py | 4 +-
tests/python/codegen/test_target_codegen_vulkan.py | 22 +--
tests/python/codegen/test_target_codegen_x86.py | 2 +-
.../test_hexagon/test_2d_physical_buffers.py | 4 +-
.../test_hexagon/test_async_dma_pipeline.py | 8 +-
.../test_hexagon/test_benchmark_maxpool2d.py | 2 +-
.../contrib/test_hexagon/test_meta_schedule.py | 8 +-
.../contrib/test_hexagon/test_parallel_hvx.py | 4 +-
.../test_hexagon/test_parallel_hvx_load_vtcm.py | 16 +--
.../contrib/test_hexagon/test_parallel_scalar.py | 4 +-
tests/python/contrib/test_hexagon/test_sigmoid.py | 2 +-
.../test_hexagon/test_software_pipeline_async.py | 2 +-
tests/python/contrib/test_hexagon/test_vtcm.py | 2 +-
.../contrib/test_hexagon/test_vtcm_bandwidth.py | 12 +-
tests/python/dlight/test_primitives.py | 2 +-
.../meta_schedule/test_meta_schedule_cost_model.py | 2 +-
.../meta_schedule/test_meta_schedule_database.py | 4 +-
...schedule_feature_extractor_per_store_feature.py | 16 +--
.../test_meta_schedule_measure_callback.py | 2 +-
.../test_meta_schedule_mma_tensorize.py | 4 +-
...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 | 8 +-
...ule_postproc_disallow_async_strided_mem_copy.py | 4 +-
...meta_schedule_postproc_disallow_dynamic_loop.py | 4 +-
..._schedule_postproc_rewrite_cooperative_fetch.py | 4 +-
.../test_meta_schedule_postproc_rewrite_layout.py | 10 +-
...e_postproc_rewrite_parallel_vectorize_unroll.py | 4 +-
...ta_schedule_postproc_rewrite_reduction_block.py | 4 +-
...est_meta_schedule_postproc_rewrite_tensorize.py | 6 +-
...meta_schedule_postproc_rewrite_unbound_block.py | 10 +-
.../test_meta_schedule_postproc_verify_gpu_code.py | 4 +-
...est_meta_schedule_postproc_verify_vtcm_limit.py | 2 +-
...eta_schedule_schedule_rule_apply_custom_rule.py | 2 +-
...test_meta_schedule_schedule_rule_auto_inline.py | 4 +-
.../test_meta_schedule_schedule_rule_mlt_intrin.py | 6 +-
.../test_meta_schedule_schedule_rule_mlt_tc.py | 2 +-
.../test_meta_schedule_search_strategy.py | 4 +-
...ule_space_cuda_async_multiple_initialization.py | 2 +-
.../test_meta_schedule_space_generator.py | 2 +-
.../test_meta_schedule_task_scheduler.py | 4 +-
.../test_meta_schedule_trace_apply.py | 7 +-
.../meta_schedule/test_meta_schedule_tune_tir.py | 2 +-
.../test_analysis_suggest_layout_transforms.py | 2 +-
tests/python/relax/test_frontend_dynamo.py | 2 +-
.../relax/test_frontend_from_exported_program.py | 25 ++++
.../python/relax/test_transform_few_shot_tuning.py | 4 +-
.../test_transform_meta_schedule_apply_database.py | 2 +-
tests/python/relax/test_vm_build.py | 2 +-
tests/python/relax/texture/test_texture_nd.py | 4 +-
.../python/runtime/test_evaluator_with_preproc.py | 2 +-
tests/python/runtime/test_runtime_module_load.py | 2 +-
tests/python/runtime/test_runtime_rpc.py | 2 +-
.../base}/test_tir_block_dependence_info.py | 5 +-
.../schedule}/test_tir_schedule_analysis.py | 12 +-
.../test_tir_schedule_annotate_buffer_access.py | 16 +--
.../schedule}/test_tir_schedule_block_scope.py | 12 +-
.../schedule}/test_tir_schedule_blockize.py | 16 +--
.../schedule}/test_tir_schedule_cache_index.py | 6 +-
.../test_tir_schedule_cache_read_write.py | 108 +++++++-------
.../schedule}/test_tir_schedule_compute_at.py | 94 ++++++------
.../schedule}/test_tir_schedule_compute_inline.py | 112 +++++++--------
.../test_tir_schedule_decompose_padding.py | 18 +--
.../schedule}/test_tir_schedule_error.py | 18 +--
.../schedule}/test_tir_schedule_for_kind.py | 56 ++++----
.../test_tir_schedule_fuse_reduction_epilogue.py | 20 +--
...ir_schedule_fuse_reduction_epilogue_clipping.py | 10 +-
...st_tir_schedule_fuse_reduction_epilogue_relu.py | 8 +-
.../schedule}/test_tir_schedule_instruction.py | 2 +-
.../schedule}/test_tir_schedule_merge.py | 26 ++--
.../schedule}/test_tir_schedule_pad_einsum.py | 8 +-
.../schedule}/test_tir_schedule_partition.py | 36 ++---
.../schedule}/test_tir_schedule_read_write_at.py | 8 +-
.../schedule}/test_tir_schedule_reduction.py | 24 ++--
.../schedule}/test_tir_schedule_reindex.py | 18 +--
.../schedule}/test_tir_schedule_reorder.py | 48 +++----
.../test_tir_schedule_reorder_block_iter_var.py | 16 +--
.../schedule}/test_tir_schedule_rfactor.py | 160 +++++++++++----------
.../schedule}/test_tir_schedule_rolling_buffer.py | 32 ++---
.../schedule}/test_tir_schedule_sampling.py | 20 +--
.../test_tir_schedule_set_axis_separator.py | 12 +-
.../schedule}/test_tir_schedule_set_dtype.py | 16 +--
.../schedule}/test_tir_schedule_set_scope.py | 20 +--
.../schedule}/test_tir_schedule_split_fuse.py | 82 +++++------
.../schedule}/test_tir_schedule_state.py | 8 +-
.../test_tir_schedule_state_cached_flags.py | 46 +++---
.../schedule}/test_tir_schedule_storage_align.py | 30 ++--
.../schedule}/test_tir_schedule_tensorize.py | 36 ++---
..._tir_schedule_tensorize_ldmatrix_mma_numeric.py | 2 +-
.../test_tir_schedule_tensorize_mfma_numeric.py | 2 +-
.../schedule}/test_tir_schedule_trace.py | 54 +++----
.../schedule}/test_tir_schedule_transform.py | 6 +-
.../test_tir_schedule_transform_layout.py | 88 ++++++------
.../schedule}/test_tir_schedule_utilities.py | 40 +++---
tests/python/te/test_te_create_primfunc.py | 16 +--
.../python/testing/test_type_annotation_checker.py | 4 +-
...test_tir_analysis_calculate_allocated_memory.py | 4 +-
.../test_tir_analysis_get_block_access_region.py | 2 +-
tests/python/tir-analysis/test_tir_analysis_oob.py | 8 +-
tests/python/tir-base/test_tir_intrin.py | 12 +-
tests/python/tir-base/test_tir_ptx_mma.py | 34 ++---
tests/python/tir-base/test_tir_ptx_mma_sp.py | 4 +-
tests/python/tir-base/test_tir_specialize.py | 2 +-
tests/python/tir-base/test_tir_texture_scope.py | 2 +-
.../tir-base/test_tir_unsafe_hide_buffer_access.py | 10 +-
.../test_tir_transform_inject_software_pipeline.py | 12 +-
.../test_tir_transform_lower_tvm_builtin.py | 2 +-
...sform_plan_update_buffer_allocation_location.py | 2 +-
.../tvmscript/test_tvmscript_error_report.py | 16 +--
.../tvmscript/test_tvmscript_syntax_sugar.py | 2 +-
tests/scripts/task_python_unittest.sh | 3 +-
web/tests/python/relax_rpc_test.py | 2 +-
web/tests/python/webgpu_rpc_test.py | 2 +-
215 files changed, 1319 insertions(+), 1408 deletions(-)
rename docs/reference/api/python/{tir => s_tir}/schedule.rst (92%)
copy tests/python/contrib/test_hexagon/conftest.py =>
python/tvm/s_tir/__init__.py (72%)
copy python/tvm/{tir => s_tir}/_ffi_api.py (100%)
copy python/tvm/{tir => s_tir}/backend/__init__.py (93%)
copy python/tvm/{contrib/hexagon => s_tir/backend/adreno}/__init__.py (92%)
copy python/tvm/{tir => s_tir}/backend/adreno/pipeline.py (84%)
rename python/tvm/{tir => s_tir}/block_dependence_info.py (100%)
rename python/tvm/{tir => s_tir}/block_scope.py (100%)
rename python/tvm/{tir/backend/adreno => s_tir}/pipeline.py (87%)
rename python/tvm/{tir => s_tir}/schedule/__init__.py (100%)
rename python/tvm/{tir => s_tir}/schedule/_ffi_api.py (95%)
rename python/tvm/{tir => s_tir}/schedule/_type_checker.py (100%)
rename python/tvm/{tir => s_tir}/schedule/analysis.py (97%)
rename python/tvm/{tir => s_tir}/schedule/instruction.py (100%)
rename python/tvm/{tir => s_tir}/schedule/schedule.py (98%)
rename python/tvm/{tir => s_tir}/schedule/state.py (100%)
rename python/tvm/{tir => s_tir}/schedule/testing.py (98%)
rename python/tvm/{tir => s_tir}/schedule/trace.py (99%)
rename python/tvm/{tir => s_tir}/schedule/transform.py (96%)
rename python/tvm/{tir => s_tir}/tensor_intrin/__init__.py (100%)
rename python/tvm/{tir => s_tir}/tensor_intrin/arm_cpu.py (100%)
rename python/tvm/{tir => s_tir}/tensor_intrin/cuda.py (100%)
rename python/tvm/{tir => s_tir}/tensor_intrin/dot_product_common.py (100%)
rename python/tvm/{tir => s_tir}/tensor_intrin/hexagon.py (100%)
rename python/tvm/{tir => s_tir}/tensor_intrin/metal.py (100%)
rename python/tvm/{tir => s_tir}/tensor_intrin/riscv_cpu.py (100%)
rename python/tvm/{tir => s_tir}/tensor_intrin/rocm.py (100%)
rename python/tvm/{tir => s_tir}/tensor_intrin/x86.py (100%)
rename tests/python/{tir-base => s_tir/base}/test_tir_block_dependence_info.py
(97%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_analysis.py (98%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_annotate_buffer_access.py (97%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_block_scope.py (93%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_blockize.py (97%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_cache_index.py (99%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_cache_read_write.py (95%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_compute_at.py (96%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_compute_inline.py (95%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_decompose_padding.py (97%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_error.py (85%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_for_kind.py (93%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_fuse_reduction_epilogue.py (93%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_fuse_reduction_epilogue_clipping.py (96%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_fuse_reduction_epilogue_relu.py (97%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_instruction.py (96%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_merge.py (92%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_pad_einsum.py (98%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_partition.py (94%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_read_write_at.py (98%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_reduction.py (95%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_reindex.py (97%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_reorder.py (90%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_reorder_block_iter_var.py (85%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_rfactor.py (92%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_rolling_buffer.py (96%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_sampling.py (92%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_set_axis_separator.py (96%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_set_dtype.py (92%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_set_scope.py (91%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_split_fuse.py (92%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_state.py (98%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_state_cached_flags.py (95%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_storage_align.py (89%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_tensorize.py (97%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_tensorize_ldmatrix_mma_numeric.py (99%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_tensorize_mfma_numeric.py (99%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_trace.py (88%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_transform.py (97%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_transform_layout.py (95%)
rename tests/python/{tir-schedule =>
s_tir/schedule}/test_tir_schedule_utilities.py (92%)
