This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a change to branch tvmscript-generic-parser-builder
in repository https://gitbox.apache.org/repos/asf/tvm.git
from b1104fd4dc Merge upstream allocation bounds and scan index width
updates
new 587d58f9ce Preserve eager and postponed TVMScript annotations without
caller migration
new d1c62b5179 Start fresh annotation symbols after interrupted definition
attempts
The 2 revisions listed above as "new" are entirely new to this
repository and will be described in separate emails. The revisions
listed as "add" were already present in the repository and have only
been added to this reference.
Summary of changes:
docs/deep_dive/relax/tutorials/relax_creation.py | 2 +-
docs/deep_dive/tensor_ir/tutorials/tir_creation.py | 3 +-
.../tensor_ir/tutorials/tir_transformation.py | 3 +-
docs/get_started/tutorials/ir_module.py | 2 -
docs/how_to/tutorials/bring_your_own_codegen.py | 4 +-
.../tutorials/mix_python_and_tvm_with_pymodule.py | 3 +-
python/tvm/backend/cuda/lang/tile_scheduler.py | 2 -
python/tvm/relax/backend/gpu_generic/cumsum.py | 2 -
python/tvm/relax/backend/gpu_generic/sampling.py | 2 -
.../tvm/relax/frontend/nn/llm/_decode_kernels.py | 1 -
python/tvm/relax/frontend/nn/llm/_kernel_common.py | 1 -
python/tvm/relax/frontend/nn/llm/_page_kernels.py | 1 -
.../tvm/relax/frontend/nn/llm/_prefill_kernels.py | 1 -
.../relax/frontend/nn/llm/position_embedding.py | 2 -
python/tvm/relax/frontend/nn/llm/tree_attn.py | 2 -
python/tvm/relax/frontend/nn/op.py | 2 -
.../tvm/relax/frontend/tflite/tflite_frontend.py | 2 -
python/tvm/relax/script/builder/distributed/ir.py | 8 +-
python/tvm/relax/script/builder_v2/__init__.py | 5 +
.../tvm/relax/transform/legalize_ops/inspect_op.py | 2 -
python/tvm/s_tir/tensor_intrin/arm_cpu.py | 2 -
python/tvm/s_tir/tensor_intrin/cuda.py | 2 -
.../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 | 2 -
python/tvm/s_tir/tensor_intrin/riscv_cpu.py | 2 -
python/tvm/s_tir/tensor_intrin/rocm.py | 2 -
python/tvm/s_tir/tensor_intrin/x86.py | 2 -
python/tvm/script/ir_builder/ir/ir.py | 73 +++++++-
python/tvm/script/parser_v2/annotations.py | 193 +++++++++++++++++++++
python/tvm/script/parser_v2/frontend.py | 55 +++++-
python/tvm/tirx/bench.py | 2 -
python/tvm/tirx/script/builder_v2/__init__.py | 5 +
tests/python/codegen/test_codegen_assert.py | 2 -
.../python/codegen/test_codegen_error_handling.py | 2 -
tests/python/codegen/test_gpu_codegen_allreduce.py | 2 -
tests/python/codegen/test_inject_ptx_ldg32.py | 2 -
tests/python/codegen/test_target_codegen.py | 2 -
.../python/codegen/test_target_codegen_aarch64.py | 2 -
tests/python/codegen/test_target_codegen_arm.py | 2 -
tests/python/codegen/test_target_codegen_blob.py | 2 -
tests/python/codegen/test_target_codegen_bool.py | 2 -
tests/python/codegen/test_target_codegen_c_host.py | 2 -
.../codegen/test_target_codegen_cross_llvm.py | 2 -
tests/python/codegen/test_target_codegen_cuda.py | 2 -
.../codegen/test_target_codegen_cuda_fastmath.py | 2 -
.../python/codegen/test_target_codegen_cuda_fp4.py | 2 -
.../python/codegen/test_target_codegen_cuda_fp8.py | 2 -
tests/python/codegen/test_target_codegen_device.py | 2 -
tests/python/codegen/test_target_codegen_extern.py | 2 -
.../codegen/test_target_codegen_gpu_common.py | 2 -
.../python/codegen/test_target_codegen_hexagon.py | 2 -
tests/python/codegen/test_target_codegen_llvm.py | 2 -
.../python/codegen/test_target_codegen_llvm_vla.py | 2 -
tests/python/codegen/test_target_codegen_metal.py | 2 -
tests/python/codegen/test_target_codegen_opencl.py | 2 -
tests/python/codegen/test_target_codegen_riscv.py | 2 -
tests/python/codegen/test_target_codegen_rocm.py | 2 -
.../codegen/test_target_codegen_static_init.py | 2 -
tests/python/codegen/test_target_codegen_vulkan.py | 2 -
tests/python/codegen/test_target_codegen_webgpu.py | 2 -
tests/python/codegen/test_target_codegen_x86.py | 2 -
.../contrib/test_android/test_meta_schedule.py | 2 -
tests/python/contrib/test_example_npu.py | 2 -
.../python/contrib/test_tir_triton_integration.py | 2 -
tests/python/disco/test_callback.py | 4 +-
tests/python/disco/test_ccl.py | 2 -
tests/python/disco/test_loader.py | 2 -
tests/python/disco/test_nvshmem.py | 2 -
tests/python/disco/test_session.py | 2 -
tests/python/driver/test_compile.py | 2 -
tests/python/ir/test_datatype_nv_fp8.py | 2 -
tests/python/ir/test_pass_instrument.py | 2 -
tests/python/relax/backend/adreno/mod_utils.py | 2 -
.../relax/backend/adreno/test_texture_network.py | 2 -
.../relax/backend/adreno/test_texture_ops.py | 2 -
.../adreno/test_transform_annotate_custom_scope.py | 2 -
.../test_transform_fold_vdevice_scope_change.py | 2 -
..._distributed_transform_legalize_redistribute.py | 2 -
.../test_distributed_transform_lower_distir.py | 2 -
...ributed_transform_lower_global_to_local_view.py | 2 -
...est_distributed_transform_propagate_sharding.py | 2 -
.../test_distributed_tvmscript_parser.py | 2 -
.../test_distributed_tvmscript_printer.py | 2 -
tests/python/relax/test_analysis.py | 2 -
.../test_analysis_computable_at_compile_time.py | 2 -
.../relax/test_analysis_contains_impure_call.py | 2 -
.../python/relax/test_analysis_detect_recursion.py | 2 -
.../relax/test_analysis_estimate_memory_usage.py | 2 -
.../test_analysis_suggest_layout_transforms.py | 2 -
tests/python/relax/test_analysis_type_analysis.py | 2 -
tests/python/relax/test_analysis_well_formed.py | 2 -
tests/python/relax/test_ast_printer.py | 2 -
.../python/relax/test_backend_dispatch_sampling.py | 2 -
.../relax/test_backend_dispatch_sort_scan.py | 2 -
.../relax/test_backend_transform_shape_lower.py | 2 -
tests/python/relax/test_base_py_module_printer.py | 2 -
.../relax/test_base_py_module_symbolic_shape.py | 2 -
tests/python/relax/test_bind_params.py | 2 -
tests/python/relax/test_bind_symbolic_vars.py | 2 -
tests/python/relax/test_binding_rewrite.py | 2 -
tests/python/relax/test_blockbuilder_core.py | 2 -
tests/python/relax/test_blockbuilder_emit_te.py | 3 +-
tests/python/relax/test_codegen_cublas.py | 2 -
tests/python/relax/test_codegen_cutlass.py | 2 -
tests/python/relax/test_codegen_dnnl.py | 2 -
tests/python/relax/test_codegen_tensorrt.py | 2 -
tests/python/relax/test_contrib_vllm.py | 2 -
tests/python/relax/test_dataflow_inplace.py | 2 -
tests/python/relax/test_dataflow_pattern.py | 2 -
tests/python/relax/test_dataflow_rewriter.py | 2 -
tests/python/relax/test_dlpack_integration.py | 2 -
tests/python/relax/test_e2e_op_dynamic.py | 2 -
...eliminate_pad_branch_using_buffer_assumption.py | 2 -
tests/python/relax/test_expr_functor.py | 2 -
tests/python/relax/test_fast_math_transform.py | 2 -
tests/python/relax/test_frontend_common.py | 2 -
tests/python/relax/test_frontend_dynamo.py | 2 -
.../relax/test_frontend_from_exported_program.py | 2 -
tests/python/relax/test_frontend_from_fx.py | 2 -
tests/python/relax/test_frontend_nn_exporter.py | 2 -
.../python/relax/test_frontend_nn_extern_module.py | 2 -
tests/python/relax/test_frontend_nn_modules.py | 2 -
tests/python/relax/test_frontend_nn_op.py | 2 -
tests/python/relax/test_frontend_nn_packing.py | 2 -
tests/python/relax/test_frontend_nn_subroutines.py | 2 -
tests/python/relax/test_frontend_nn_tensor.py | 2 -
tests/python/relax/test_frontend_stablehlo.py | 2 -
tests/python/relax/test_frontend_tflite.py | 2 -
tests/python/relax/test_inline_functions.py | 2 -
tests/python/relax/test_kill_after_last_use.py | 2 -
.../relax/test_meta_schedule_relax_integration.py | 2 -
tests/python/relax/test_op_image.py | 2 -
tests/python/relax/test_op_index.py | 2 -
tests/python/relax/test_op_inspect.py | 2 -
tests/python/relax/test_op_misc.py | 2 -
tests/python/relax/test_op_size.py | 2 -
tests/python/relax/test_op_take.py | 2 -
tests/python/relax/test_op_view.py | 2 -
tests/python/relax/test_op_vision.py | 2 -
.../python/relax/test_optimize_layout_transform.py | 2 -
tests/python/relax/test_pipeline.py | 2 -
tests/python/relax/test_pytorch_integration.py | 2 -
tests/python/relax/test_relax_operators.py | 2 -
.../python/relax/test_relax_to_pyfunc_converter.py | 2 -
.../python/relax/test_remove_redundant_reshape.py | 2 -
.../python/relax/test_runtime_builtin_rnn_state.py | 2 -
tests/python/relax/test_testing_nn.py | 2 -
tests/python/relax/test_tir_call_source_kernel.py | 2 -
tests/python/relax/test_training_append_loss.py | 2 -
tests/python/relax/test_training_loss.py | 2 -
tests/python/relax/test_training_optimizer.py | 2 -
tests/python/relax/test_training_setup_trainer.py | 2 -
.../python/relax/test_training_trainer_numeric.py | 2 -
tests/python/relax/test_transform.py | 2 -
.../relax/test_transform_adjust_matmul_order.py | 2 -
.../relax/test_transform_allocate_workspace.py | 2 -
tests/python/relax/test_transform_alter_op_impl.py | 2 -
.../test_transform_annotate_tir_op_pattern.py | 2 -
...st_transform_attach_attr_layout_free_buffers.py | 2 -
.../relax/test_transform_attach_global_symbol.py | 2 -
tests/python/relax/test_transform_bind_params.py | 2 -
.../relax/test_transform_bind_symbolic_vars.py | 2 -
.../relax/test_transform_bundle_model_params.py | 2 -
.../relax/test_transform_canonicalize_bindings.py | 2 -
tests/python/relax/test_transform_codegen_pass.py | 2 -
.../test_transform_combine_parallel_matmul.py | 2 -
.../relax/test_transform_compute_prim_value.py | 2 -
.../relax/test_transform_convert_dataflow.py | 2 -
.../python/relax/test_transform_convert_layout.py | 2 -
tests/python/relax/test_transform_cse.py | 2 -
.../relax/test_transform_dead_code_elimination.py | 2 -
tests/python/relax/test_transform_decompose_ops.py | 2 -
.../relax/test_transform_expand_matmul_of_sum.py | 2 -
.../relax/test_transform_expand_tuple_args.py | 2 -
tests/python/relax/test_transform_fold_constant.py | 2 -
tests/python/relax/test_transform_fuse_ops.py | 2 -
.../relax/test_transform_fuse_ops_by_pattern.py | 2 -
tests/python/relax/test_transform_fuse_tir.py | 2 -
.../relax/test_transform_fuse_transpose_matmul.py | 2 -
tests/python/relax/test_transform_gradient.py | 2 -
.../relax/test_transform_gradient_checkpoint.py | 2 -
.../relax/test_transform_gradient_numeric.py | 2 -
.../relax/test_transform_gradient_te_register.py | 2 -
.../test_transform_inline_private_functions.py | 2 -
.../relax/test_transform_ipc_allreduce_rewrite.py | 2 -
tests/python/relax/test_transform_lambda_lift.py | 2 -
.../relax/test_transform_lazy_transform_params.py | 2 -
tests/python/relax/test_transform_legalize_ops.py | 2 -
.../relax/test_transform_legalize_ops_binary.py | 2 -
.../relax/test_transform_legalize_ops_ccl.py | 2 -
.../test_transform_legalize_ops_create_datatype.py | 2 -
.../test_transform_legalize_ops_distributed.py | 2 -
.../relax/test_transform_legalize_ops_grad.py | 2 -
.../relax/test_transform_legalize_ops_image.py | 2 -
..._transform_legalize_ops_index_linear_algebra.py | 2 -
.../test_transform_legalize_ops_manipulate.py | 2 -
.../python/relax/test_transform_legalize_ops_nn.py | 2 -
.../relax/test_transform_legalize_ops_qdq.py | 2 -
...st_transform_legalize_ops_search_statistical.py | 2 -
.../relax/test_transform_legalize_ops_unary.py | 2 -
.../relax/test_transform_lift_transform_params.py | 2 -
.../test_transform_lower_gpu_ipc_alloc_storage.py | 2 -
.../test_transform_merge_composite_functions.py | 2 -
.../test_transform_meta_schedule_apply_database.py | 2 -
.../relax/test_transform_meta_schedule_tuning.py | 2 -
tests/python/relax/test_transform_normalize.py | 2 -
.../relax/test_transform_normalize_global_var.py | 2 -
...st_transform_operator_specific_normalization.py | 2 -
.../python/relax/test_transform_realize_vdevice.py | 2 -
.../relax/test_transform_remove_unused_outputs.py | 2 -
.../test_transform_remove_unused_parameters.py | 2 -
..._transform_reorder_permute_dims_after_concat.py | 2 -
.../test_transform_reorder_take_after_matmul.py | 2 -
.../relax/test_transform_rewrite_cuda_graph.py | 2 -
.../test_transform_rewrite_dataflow_reshape.py | 2 -
...nsform_specialize_primfunc_based_on_callsite.py | 2 -
.../test_transform_split_layout_rewrite_preproc.py | 2 -
.../test_transform_static_plan_block_memory.py | 2 -
.../relax/test_transform_to_mixed_precision.py | 2 -
.../relax/test_transform_topological_sort.py | 2 -
.../relax/test_transform_update_param_type.py | 2 -
.../python/relax/test_transform_update_vdevice.py | 2 -
tests/python/relax/test_tvmscript_ir_builder.py | 2 -
tests/python/relax/test_tvmscript_parser.py | 2 -
.../relax/test_tvmscript_parser_op_arith_cmp.py | 2 -
.../relax/test_tvmscript_parser_op_create.py | 2 -
.../relax/test_tvmscript_parser_op_datatype.py | 2 -
.../python/relax/test_tvmscript_parser_op_grad.py | 2 -
.../python/relax/test_tvmscript_parser_op_image.py | 2 -
.../python/relax/test_tvmscript_parser_op_index.py | 2 -
.../test_tvmscript_parser_op_linear_algebra.py | 2 -
.../relax/test_tvmscript_parser_op_manipulate.py | 2 -
tests/python/relax/test_tvmscript_parser_op_nn.py | 2 -
.../relax/test_tvmscript_parser_op_search.py | 2 -
tests/python/relax/test_tvmscript_parser_op_set.py | 2 -
.../python/relax/test_tvmscript_parser_op_sort.py | 2 -
.../relax/test_tvmscript_parser_op_statistical.py | 2 -
.../relax/test_tvmscript_parser_op_vision.py | 2 -
tests/python/relax/test_tvmscript_printer_relax.py | 2 -
tests/python/relax/test_tvmscript_pyfunc.py | 2 -
tests/python/relax/test_tvmscript_type_vars.py | 2 -
tests/python/relax/test_utils.py | 2 -
.../relax/test_vm_alloc_storage_with_scope.py | 2 -
tests/python/relax/test_vm_build.py | 2 -
tests/python/relax/test_vm_builtin.py | 2 -
tests/python/relax/test_vm_builtin_lower.py | 2 -
tests/python/relax/test_vm_callback_function.py | 2 -
tests/python/relax/test_vm_codegen_only.py | 2 -
tests/python/relax/test_vm_codegen_tir.py | 2 -
tests/python/relax/test_vm_cuda_graph.py | 2 -
tests/python/relax/test_vm_execbuilder.py | 2 -
tests/python/relax/test_vm_multi_device.py | 2 -
tests/python/relax/texture/test_texture_nd.py | 2 -
tests/python/runtime/test_executable.py | 2 -
tests/python/runtime/test_runtime_extension.py | 2 -
tests/python/runtime/test_runtime_rpc.py | 2 -
...st_s_tir_analysis_calculate_allocated_memory.py | 2 -
.../test_s_tir_analysis_estimate_tir_flops.py | 2 -
.../test_s_tir_analysis_identify_memcpy.py | 2 -
.../test_s_tir_analysis_is_pure_function.py | 2 -
.../s_tir/analysis/test_s_tir_analysis_oob.py | 2 -
.../s_tir/analysis/test_sblock_access_region.py | 2 -
.../analysis/test_sblock_buffer_access_lca.py | 2 -
.../s_tir/base/test_sblock_dependence_info.py | 2 -
.../s_tir/base/test_tir_te_extern_primfunc.py | 2 -
tests/python/s_tir/dlight/test_benchmark.py | 2 -
tests/python/s_tir/dlight/test_cpu_gemv.py | 2 -
tests/python/s_tir/dlight/test_gpu_conv.py | 2 -
tests/python/s_tir/dlight/test_gpu_fallback.py | 2 -
tests/python/s_tir/dlight/test_gpu_gemv.py | 2 -
.../s_tir/dlight/test_gpu_general_reduction.py | 2 -
.../python/s_tir/dlight/test_gpu_low_batch_gemv.py | 2 -
tests/python/s_tir/dlight/test_gpu_matmul.py | 2 -
.../s_tir/dlight/test_gpu_matmul_tensorize.py | 2 -
tests/python/s_tir/dlight/test_gpu_reduction.py | 2 -
tests/python/s_tir/dlight/test_gpu_rmsnorm.py | 2 -
tests/python/s_tir/dlight/test_gpu_transpose.py | 2 -
tests/python/s_tir/dlight/test_primitives.py | 2 -
.../meta_schedule/test_meta_schedule_arg_info.py | 2 -
.../meta_schedule/test_meta_schedule_builder.py | 2 -
.../meta_schedule/test_meta_schedule_cost_model.py | 8 +-
.../meta_schedule/test_meta_schedule_database.py | 2 -
...schedule_feature_extractor_per_store_feature.py | 2 -
.../test_meta_schedule_measure_callback.py | 2 -
.../test_meta_schedule_mma_tensorize.py | 2 -
...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 | 12 +-
...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 | 2 -
...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 | 2 -
...est_meta_schedule_postproc_verify_vtcm_limit.py | 2 -
.../meta_schedule/test_meta_schedule_runner.py | 2 -
...test_meta_schedule_schedule_rule_add_rfactor.py | 2 -
...eta_schedule_schedule_rule_apply_custom_rule.py | 2 -
.../test_meta_schedule_schedule_rule_auto_bind.py | 2 -
...test_meta_schedule_schedule_rule_auto_inline.py | 2 -
...chedule_schedule_rule_cross_thread_reduction.py | 2 -
.../test_meta_schedule_schedule_rule_mlt.py | 2 -
.../test_meta_schedule_schedule_rule_mlt_intrin.py | 2 -
.../test_meta_schedule_schedule_rule_mlt_tc.py | 2 -
...dule_schedule_rule_parallel_vectorize_unroll.py | 2 -
...hedule_schedule_rule_random_compute_location.py | 2 -
.../test_meta_schedule_search_strategy.py | 5 +-
.../meta_schedule/test_meta_schedule_space_cpu.py | 2 -
.../meta_schedule/test_meta_schedule_space_cuda.py | 2 -
.../test_meta_schedule_space_cuda_async.py | 2 -
.../test_meta_schedule_space_generator.py | 4 +-
.../test_meta_schedule_space_post_opt.py | 2 -
.../test_meta_schedule_task_scheduler.py | 4 +-
.../test_meta_schedule_trace_apply.py | 2 -
.../test_meta_schedule_tune_context.py | 2 -
.../meta_schedule/test_meta_schedule_tune_tir.py | 4 +-
.../s_tir/schedule/test_tir_schedule_analysis.py | 2 -
.../test_tir_schedule_annotate_buffer_access.py | 2 -
.../schedule/test_tir_schedule_block_scope.py | 2 -
.../s_tir/schedule/test_tir_schedule_blockize.py | 2 -
.../schedule/test_tir_schedule_cache_index.py | 2 -
.../schedule/test_tir_schedule_cache_read_write.py | 2 -
.../s_tir/schedule/test_tir_schedule_compute_at.py | 2 -
.../schedule/test_tir_schedule_compute_inline.py | 2 -
.../test_tir_schedule_decompose_padding.py | 2 -
.../s_tir/schedule/test_tir_schedule_error.py | 2 -
.../s_tir/schedule/test_tir_schedule_for_kind.py | 2 -
.../test_tir_schedule_fuse_reduction_epilogue.py | 2 -
...ir_schedule_fuse_reduction_epilogue_clipping.py | 2 -
...st_tir_schedule_fuse_reduction_epilogue_relu.py | 2 -
.../s_tir/schedule/test_tir_schedule_merge.py | 2 -
.../s_tir/schedule/test_tir_schedule_pad_einsum.py | 2 -
.../s_tir/schedule/test_tir_schedule_partition.py | 2 -
.../schedule/test_tir_schedule_read_write_at.py | 2 -
.../s_tir/schedule/test_tir_schedule_reduction.py | 2 -
.../s_tir/schedule/test_tir_schedule_reindex.py | 2 -
.../s_tir/schedule/test_tir_schedule_reorder.py | 2 -
.../test_tir_schedule_reorder_block_iter_var.py | 2 -
.../s_tir/schedule/test_tir_schedule_rfactor.py | 2 -
.../schedule/test_tir_schedule_rolling_buffer.py | 2 -
.../s_tir/schedule/test_tir_schedule_sampling.py | 2 -
.../s_tir/schedule/test_tir_schedule_set_dtype.py | 2 -
.../s_tir/schedule/test_tir_schedule_set_scope.py | 2 -
.../s_tir/schedule/test_tir_schedule_split_fuse.py | 2 -
.../s_tir/schedule/test_tir_schedule_state.py | 2 -
.../test_tir_schedule_state_cached_flags.py | 2 -
.../schedule/test_tir_schedule_storage_align.py | 2 -
.../s_tir/schedule/test_tir_schedule_tensorize.py | 2 -
.../s_tir/schedule/test_tir_schedule_trace.py | 2 -
.../s_tir/schedule/test_tir_schedule_transform.py | 2 -
.../schedule/test_tir_schedule_transform_layout.py | 2 -
.../s_tir/schedule/test_tir_schedule_utilities.py | 2 -
tests/python/s_tir/test_arith_domain_touched.py | 2 -
tests/python/s_tir/test_s_tir_renew_defs.py | 2 -
...test_s_tir_transform_annotate_irregular_loop.py | 2 -
.../test_s_tir_transform_canonicalize_loop.py | 2 -
.../test_s_tir_transform_compact_buffer_region.py | 2 -
...est_s_tir_transform_convert_blocks_to_opaque.py | 2 -
.../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 | 2 -
.../test_s_tir_transform_inject_double_buffer.py | 2 -
.../test_s_tir_transform_inject_permuted_layout.py | 2 -
.../test_s_tir_transform_inject_ptx_async_copy.py | 2 -
.../test_s_tir_transform_inject_ptx_ldg32.py | 2 -
...est_s_tir_transform_inject_software_pipeline.py | 2 -
.../test_s_tir_transform_inject_virtual_thread.py | 2 -
.../test_s_tir_transform_lift_thread_binding.py | 2 -
.../test_s_tir_transform_loop_partition.py | 2 -
...s_tir_transform_lower_cross_thread_reduction.py | 2 -
.../test_s_tir_transform_lower_init_block.py | 2 -
.../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 | 2 -
...transform_manifest_shared_memory_local_stage.py | 2 -
...st_s_tir_transform_memhammer_lower_auto_copy.py | 2 -
...form_merge_dynamic_shared_memory_allocations.py | 2 -
...sform_plan_update_buffer_allocation_location.py | 2 -
.../test_s_tir_transform_profiling_instr.py | 2 -
.../transform/test_s_tir_transform_remove_undef.py | 2 -
...transform_remove_weight_layout_rewrite_block.py | 2 -
...st_s_tir_transform_renormalize_split_pattern.py | 2 -
.../test_s_tir_transform_rewrite_unsafe_select.py | 2 -
.../transform/test_s_tir_transform_thread_sync.py | 2 -
.../test_s_tir_transform_unify_thread_binding.py | 2 -
tests/python/target/test_arm_target.py | 2 -
tests/python/te/test_te_create_primfunc.py | 2 -
.../test_tir_analysis_verify_well_formed.py | 2 -
tests/python/tirx-base/test_tir_base.py | 2 -
tests/python/tirx-base/test_tir_host_func.py | 2 -
tests/python/tirx-base/test_tir_imm_values.py | 2 -
tests/python/tirx-base/test_tir_intrin.py | 2 -
tests/python/tirx-base/test_tir_ptx_cp_async.py | 2 -
.../tirx-base/test_tir_ptx_griddepcontrol.py | 2 -
tests/python/tirx-base/test_tir_ptx_ldmatrix.py | 2 -
tests/python/tirx-base/test_tir_ptx_mma.py | 2 -
tests/python/tirx-base/test_tir_ptx_mma_sp.py | 2 -
.../tirx-base/test_tir_ptx_scalar_f32_math.py | 2 -
tests/python/tirx-base/test_tir_specialize.py | 2 -
.../tirx-base/test_tir_stmt_functor_substitute.py | 2 -
tests/python/tirx-base/test_tir_texture_scope.py | 2 -
.../test_tir_unsafe_hide_buffer_access.py | 2 -
.../test_tir_inline_private_functions.py | 2 -
.../test_tir_transform_bf16_legalize.py | 2 -
.../test_tir_transform_common_subexpr_elim.py | 2 -
.../test_tir_transform_convert_ssa.py | 2 -
.../test_tir_transform_flatten_buffer.py | 2 -
...test_tir_transform_force_narrow_index_to_i32.py | 2 -
.../test_tir_transform_fp8_legalize.py | 2 -
.../tirx-transform/test_tir_transform_helpers.py | 2 -
.../test_tir_transform_lower_tvm_builtin.py | 2 -
.../test_tir_transform_make_packed_api.py | 2 -
.../test_tir_transform_narrow_datatype.py | 2 -
...est_tir_transform_pointer_value_type_rewrite.py | 2 -
.../test_tir_transform_remove_assume.py | 2 -
.../test_tir_transform_remove_no_op.py | 2 -
.../tirx-transform/test_tir_transform_simplify.py | 2 -
.../test_tir_transform_split_host_device.py | 2 -
.../test_tir_transform_storage_rewrite.py | 2 -
.../test_tir_transform_unroll_loop.py | 2 -
.../tirx-transform/test_tir_transform_vectorize.py | 2 -
tests/python/tirx/codegen/test_codegen_ampere.py | 2 -
.../python/tirx/codegen/test_codegen_blackwell.py | 2 -
tests/python/tirx/codegen/test_codegen_cuda.py | 2 -
tests/python/tirx/codegen/test_codegen_dsmem.py | 2 -
tests/python/tirx/codegen/test_codegen_hopper.py | 2 -
tests/python/tirx/codegen/test_codegen_nki.py | 2 -
tests/python/tirx/codegen/test_codegen_nvshmem.py | 2 -
tests/python/tirx/codegen/test_cuda_cta_reduce.py | 2 -
tests/python/tirx/codegen/test_cuda_wait_until.py | 2 -
tests/python/tirx/codegen/test_cuda_warp_reduce.py | 2 -
tests/python/tirx/codegen/test_ptx_addr.py | 2 -
tests/python/tirx/codegen/test_ptx_dialect.py | 2 -
tests/python/tirx/codegen/test_ptx_ld_st_ops.py | 2 -
.../codegen/test_tcgen05_descriptor_encoders.py | 2 -
tests/python/tirx/iket/iket_profile_workload.py | 2 -
tests/python/tirx/iket/test_iket_profiler.py | 2 -
.../tile_primitive/cuda/copy/test_fallback.py | 2 -
.../tile_primitive/cuda/copy/test_gmem_smem.py | 2 -
.../tile_primitive/cuda/copy/test_ld_stmatrix.py | 2 -
.../operator/tile_primitive/cuda/copy/test_reg.py | 2 -
.../cuda/copy/test_vec_forced_cache.py | 2 -
.../tile_primitive/cuda/copy_async/test_dsmem.py | 2 -
.../tile_primitive/cuda/copy_async/test_ldgsts.py | 2 -
.../cuda/copy_async/test_tcgen05_cp.py | 2 -
.../cuda/copy_async/test_tcgen05_ldst.py | 2 -
.../tile_primitive/cuda/copy_async/test_tma.py | 2 -
.../tile_primitive/cuda/elementwise/test_binary.py | 2 -
.../tile_primitive/cuda/elementwise/test_fma.py | 2 -
.../tile_primitive/cuda/elementwise/test_unary.py | 2 -
.../cuda/gemm/test_gemm_mma_m16n8k_.py | 2 -
.../cuda/gemm_async/test_gemm_async.py | 2 -
.../cuda/reduction/test_reduction.py | 2 -
.../operator/tile_primitive/trn/test_binary_trn.py | 2 -
.../tile_primitive/trn/test_compose_op_trn.py | 2 -
.../operator/tile_primitive/trn/test_copy_trn.py | 2 -
.../operator/tile_primitive/trn/test_gemm_trn.py | 2 -
.../tile_primitive/trn/test_private_alloc_trn.py | 2 -
.../tile_primitive/trn/test_reduction_trn.py | 2 -
.../operator/tile_primitive/trn/test_select_trn.py | 2 -
.../operator/tile_primitive/trn/test_unary_trn.py | 2 -
tests/python/tirx/test_buffer_print.py | 2 -
tests/python/tirx/test_control_flow.py | 2 -
tests/python/tirx/test_hint.py | 2 -
tests/python/tirx/test_inline.py | 2 -
tests/python/tirx/test_op_namespace_cleanup.py | 2 -
tests/python/tirx/test_parser_printer.py | 2 -
tests/python/tirx/test_printer_tir_namespaces.py | 2 -
tests/python/tirx/test_roundtrip_namespaces.py | 2 -
tests/python/tirx/test_verifier.py | 2 -
.../tirx/transform/test_transform_lower_tirx.py | 2 -
.../transform/test_transform_naive_allocator.py | 2 -
tests/python/tvmscript/test_parser_v2.py | 114 +++++++++++-
tests/python/tvmscript/test_tvmscript_complete.py | 2 -
.../tvmscript/test_tvmscript_error_report.py | 2 -
.../tvmscript/test_tvmscript_meta_programming.py | 2 -
tests/python/tvmscript/test_tvmscript_ops.py | 2 -
.../tvmscript/test_tvmscript_parser_source.py | 2 -
.../python/tvmscript/test_tvmscript_parser_tir.py | 2 -
.../tvmscript/test_tvmscript_printer_highlight.py | 2 -
.../tvmscript/test_tvmscript_printer_metadata.py | 2 -
.../test_tvmscript_printer_structural_equal.py | 2 -
.../python/tvmscript/test_tvmscript_printer_tir.py | 2 -
.../test_tvmscript_printer_underlining.py | 2 -
.../python/tvmscript/test_tvmscript_regression.py | 2 -
tests/python/tvmscript/test_tvmscript_roundtrip.py | 2 -
.../tvmscript/test_tvmscript_syntax_sugar.py | 2 -
tests/python/tvmscript/test_tvmscript_type.py | 2 -
495 files changed, 464 insertions(+), 994 deletions(-)
