This is an automated email from the ASF dual-hosted git repository.
ruihangl pushed a change to branch unity-staging
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 3de77f8def Merge remote-tracking branch 'main' into unity
add e7d12be07a [Unity][Training] Support intermediate vars as
require_grads for Gradient pass (#16011)
add e506bffa72 [Unity] Implement FNormalize attribute for operators
(#16067)
add eb20534f18 [Unity][Bugfix] Track variable usage from input to impure
functions (#16036)
add 8e8799d709 [Unity][Dlight] Enhance matmul tensorizer with Int8 support
(#16084)
add 6ce16026f7 [Unity] support symbolic var in RewriteDataflowReshape
(#16086)
add 9eeb5bcbcb [Unity] Support Regular expression matching in globalvar
dataflow pattern (#16085)
add 9100a8e8c9 [Unity] [LiftTransformParams] Treat symbolic var in weight
shape as constant (#16049)
add e37165f40f [Unity][MSC][M1.5-1.7] Add Runner and test with torch,
tensorflow && tensorrt (#16072)
add 6b9c277bf6 [Unity][Training] Simplify matmul patterns after gradient
(#16082)
add e359e7a210 [Disco] Add loader for presharded params. (#15957)
add 384f9b69ea [Unity] Add `axis` field to scatter_from_worker0 (#16092)
add 276b4cedbd [Unity][Fix] Fix `topi.rms_norm` with float32 upscale
(#16099)
add 171ef61797 [Unity] make LazyTransformParam more general (#16088)
add 7dd248b0be [Unity][Dlight] Choose perfect spatial factor in reduction
rule (#16101)
add bc10f769d8 [Unity] Add LoadParamOnWorker0 function in shard loader
(#16093)
add a3d9108050 [Smallfix][WEB] Change memory manager import for web
(#16107)
add 06a48996cc [Unity][Fix] Fix `rms_norm` tests (#16109)
add 7892af0d46 [Unity][WebGPU] Allow lower max storage buffer binding size
(#16108)
add 7a0c3f9e05 [Unity][Support] PagedKVCache support growth control
(#16112)
add 835bc82665 [Unity][TVMJS] Add md5sum to weight shards (#16122)
add d5daa9806d [Unity] Allow Customized Pipeline in `relax.build` (#16121)
add 26a20ee638 [Unity] Improved error message in relax::Normalizer (#16114)
add e7c7314046 [Unity][Transform] Keep R.ExternFunc in dead-code
elimination (#16118)
add 6f650db042 [Unity][DistIR] Legalize redistribute (#16098)
add c9de001490 [Unity] [Transform] Skip constants in CSE pass (#16125)
add 0ddfc657db [Unity] Implement FNormalize for relax.op.call_tir (#16068)
add 684a8ca6a4 [Unity][DLight] Enhance the inline consumer rule (#16124)
add bc2415bcc8 [Unity][BYOC] Support IGEMM in cuBLASLt (#16134)
add 03f7b3b852 [Bugfix][VTA] Fix FSIM compile error on macOS (#14655)
add a302e0fd63 [TIR] Enhance Python Type Annotations for TIR stmt (#16076)
add df5d3b58b8 [Fix][TIR] Fix dtype issues for match_buffer and ramp node
(#16051)
add a920a8c4dc [DOCs] Update release_process.rst about version number
modification (#16024)
add 2c2bb5e2f0 [TOPI] Reduce code redundancy in conv2d weights
transformation (#16080)
add 641225a5c5 [Fix] Broken Windows Build with LLVM (#16081)
add 2f20264b1f [Pylint] fix pylint issues from test_random to test_tedd
(#16065)
add 448de895ba [Target] Support obtain l2 cache size from target (#16039)
add db4290b608 [TIR] Support more mma intrinsics and
`get_mma_intrin_group` utility (#16073)
add 3f3473e2d5 [TIR] Enhance Python Type Annotations for TIR Expr (#16083)
add 47a0b4558e Add _ffi_api.py under script folder (#16087)
add d7e1f8e7e8 [Packaging] Include BYOC dynamic libraries into wheel
(#16034)
add afd1f24bf6 [DOCs] Fix error on linting by adding `--rev` argument
(#16089)
add 50597ce25b [DOCs] Add missing backtick to contribute/code_guide.rst
(#16097)
add 99336c34b9 [TFLite][Frontend] Support quantized SQUARE (#15914)
add 42de91ff45 [Fix] Fix `topi.rms_norm` with float32 upscale (#16091)
add f6634afc16 Add missing backtick (#15968)
add f9ac3b98b1 [Relay][Pytorch] Add support for aten::swapaxes operator
(#16079)
add 95d769e738 [Relay][Pytorch] Add support for `aten::bitwise_and`
(#16105)
add 5f5d90a3f5 [TOPI] Enhance `topi.nn.matmul` (#16052)
add 468bf2da79 [TIR][Transform] Introduce new `InjectPermutedLayout` pass
(#16070)
add c8ef902a75 [CI] Fix broken links about Jenkins (#16055)
add 707492a677 Revert "[#15157][Rust][Doc] Re-enable the Rust
documentation build (#15213)" (#16127)
add 015da7c185 [TOPI][ADRENO] Add conv2d transpose nchw texture schedule
(#15786)
add bd67d2e5eb [CI] Refactor unittest folder (#16110)
add 748882aae7 [Runtime] Parallel-for with threading backend (#16133)
new 2dbab710e8 Merge branch 'main' into 'unity'
The 1 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:
.github/workflows/main.yml | 6 +-
CMakeLists.txt | 1 +
ci/README.md | 16 +-
ci/jenkins/README.md | 4 +-
cmake/utils/FindLLVM.cmake | 1 +
conftest.py | 2 +-
docs/contribute/ci.rst | 2 +-
docs/contribute/code_guide.rst | 2 +-
docs/contribute/pull_request.rst | 4 +-
docs/contribute/release_process.rst | 16 +-
docs/reference/api/links.rst | 1 -
include/tvm/ir/expr.h | 4 +-
include/tvm/relax/analysis.h | 34 ++
include/tvm/relax/attrs/ccl.h | 12 +-
include/tvm/relax/block_builder.h | 43 ++
include/tvm/relax/distributed/transform.h | 7 +
include/tvm/relax/op_attr_types.h | 27 +-
include/tvm/runtime/threading_backend.h | 70 +++
include/tvm/target/tag.h | 11 +
include/tvm/tir/stmt.h | 8 +-
include/tvm/tir/var.h | 2 +-
include/tvm/topi/nn/rms_norm.h | 17 +-
python/setup.py | 5 +
python/tvm/_ffi/libinfo.py | 18 +-
python/tvm/contrib/msc/core/frontend/translate.py | 3 +-
python/tvm/contrib/msc/core/runtime/runner.py | 34 +-
python/tvm/contrib/msc/core/transform/transform.py | 18 +
python/tvm/contrib/msc/core/utils/file.py | 8 +-
python/tvm/contrib/msc/core/utils/info.py | 8 +-
.../msc/framework/tensorflow/frontend/translate.py | 3 +-
.../msc/framework/tensorflow/runtime}/__init__.py | 4 +-
.../msc/framework/tensorflow/runtime/runner.py | 217 ++++++++
.../msc/framework/tensorrt/codegen/codegen.py | 4 +-
.../msc/framework/tensorrt/runtime}/__init__.py | 4 +-
.../msc/framework/tensorrt/runtime/runner.py | 39 +-
.../msc/framework/torch/runtime}/__init__.py | 4 +-
.../contrib/msc/framework/torch/runtime/runner.py | 197 +++++++
.../contrib/msc/framework/tvm/runtime/runner.py | 8 +-
python/tvm/contrib/tvmjs.py | 22 +-
python/tvm/dlight/gpu/matmul.py | 221 +++++++-
python/tvm/dlight/gpu/reduction.py | 10 +-
python/tvm/ir/expr.py | 41 +-
python/tvm/relax/backend/contrib/cublas.py | 17 +-
python/tvm/relax/distributed/transform/__init__.py | 2 +-
.../tvm/relax/distributed/transform/transform.py | 13 +
python/tvm/relax/ir/instrument.py | 19 +-
python/tvm/relax/op/_op_gradient.py | 8 +-
python/tvm/relax/op/ccl/ccl.py | 7 +-
python/tvm/relax/op/distributed/__init__.py | 2 +-
python/tvm/relax/op/distributed/distributed.py | 26 +
python/tvm/relax/op/manipulate.py | 2 +-
python/tvm/relax/pipeline.py | 32 +-
python/tvm/relax/testing/matmul.py | 11 +-
python/tvm/relax/testing/transform.py | 6 +
.../tvm/relax/transform/lazy_transform_params.py | 117 ++++-
.../tvm/relax/transform/legalize_ops/__init__.py | 1 +
python/tvm/relax/transform/legalize_ops/ccl.py | 45 +-
.../relax/transform/legalize_ops/distributed.py | 43 ++
python/tvm/relax/vm_build.py | 30 +-
python/tvm/relay/frontend/pytorch.py | 10 +
python/tvm/relay/frontend/tflite.py | 5 -
python/tvm/relay/op/nn/_nn.py | 6 +
python/tvm/relay/op/strategy/adreno.py | 52 ++
.../transform/__init__.py => script/_ffi_api.py} | 7 +-
.../tvm/script/ir_builder/relax/distributed/ir.py | 3 +-
python/tvm/script/ir_builder/tir/ir.py | 3 +-
python/tvm/script/parser/relax/__init__.py | 27 +-
python/tvm/script/parser/relax/dist.py | 8 +-
python/tvm/testing/plugin.py | 2 +-
python/tvm/tir/expr.py | 379 +++++++++-----
python/tvm/tir/op.py | 1 -
python/tvm/tir/stmt.py | 298 ++++++++---
python/tvm/tir/tensor_intrin/cuda.py | 571 ++++++++++++++-------
python/tvm/topi/adreno/__init__.py | 2 +
.../tvm/topi/adreno/conv2d_transpose_alter_op.py | 121 +++++
python/tvm/topi/adreno/conv2d_transpose_nchw.py | 412 +++++++++++++++
python/tvm/topi/adreno/utils.py | 23 +
python/tvm/topi/arm_cpu/arm_utils.py | 38 ++
python/tvm/topi/arm_cpu/conv2d_alter_op.py | 18 +-
python/tvm/topi/nn/conv2d.py | 38 +-
python/tvm/topi/nn/dense.py | 125 +++--
python/tvm/topi/testing/rms_norm_python.py | 9 +-
src/auto_scheduler/transform_step.cc | 2 +-
src/contrib/msc/core/codegen/codegen_utils.cc | 2 +-
src/contrib/msc/core/ir/graph.cc | 3 +
src/contrib/msc/core/ir/graph_builder.cc | 22 +-
src/contrib/msc/core/printer/prototxt_printer.cc | 6 +-
src/contrib/msc/core/transform/set_byoc_attrs.cc | 92 ++++
src/contrib/msc/framework/tensorflow/codegen.cc | 3 +
.../msc/framework/tensorflow/tf_v1_opcode.cc | 46 +-
src/contrib/msc/framework/tensorrt/codegen.cc | 7 +-
src/contrib/msc/framework/tensorrt/codegen_utils.h | 2 +-
src/relax/analysis/tir_op_pattern_kind.cc | 13 +-
src/relax/analysis/udchain.cc | 88 ++--
src/relax/analysis/well_formed.cc | 53 +-
.../distributed/transform/legalize_redistribute.cc | 123 +++++
src/relax/ir/binding_rewrite.cc | 16 +-
src/relax/ir/block_builder.cc | 42 +-
src/relax/ir/dataflow_matcher.cc | 7 +-
src/relax/op/ccl/ccl.cc | 17 +-
src/relax/op/ccl/ccl.h | 2 +-
src/relax/op/distributed/distributed.cc | 82 +++
src/relax/op/distributed/distributed.h | 12 +
src/relax/op/op.cc | 88 +++-
src/relax/{op/ccl/ccl.h => testing/transform.cc} | 37 +-
src/relax/transform/dead_code_elimination.cc | 2 +-
src/relax/transform/eliminate_common_subexpr.cc | 2 +-
src/relax/transform/gradient.cc | 116 +++--
src/relax/transform/gradient_simplifier.cc | 200 ++++++++
.../ccl/ccl.h => transform/gradient_simplifier.h} | 34 +-
src/relax/transform/lift_transform_params.cc | 19 +-
src/relay/transforms/annotate_texture_storage.cc | 4 +
src/runtime/contrib/cublas/cublas.cc | 18 +-
src/runtime/disco/loader.cc | 164 +++++-
src/runtime/disco/nccl/nccl.cc | 2 +-
src/runtime/disco/worker.cc | 2 -
src/runtime/hexagon/profiler/README.md | 2 +-
src/runtime/relax_vm/paged_kv_cache.cc | 46 +-
src/script/ir_builder/relax/ir.cc | 3 +-
src/support/ordered_set.h | 7 +
src/target/tag.cc | 7 +-
src/target/target_kind.cc | 1 +
src/tir/ir/expr.cc | 4 +-
src/tir/transforms/inject_permuted_layout.cc | 401 ++++++++-------
src/topi/nn.cc | 2 +-
tests/cpp/threading_backend_test.cc | 9 +
tests/lint/pylint.sh | 12 +-
.../test_arith_canonical_simplify.py | 0
.../test_arith_const_int_bound.py | 0
.../{unittest => arith}/test_arith_deduce_bound.py | 0
.../test_arith_detect_clip_bound.py | 0
.../{unittest => arith}/test_arith_detect_cse.py | 0
.../test_arith_detect_linear_equation.py | 0
.../test_arith_domain_touched.py | 0
.../{unittest => arith}/test_arith_intset.py | 0
.../test_arith_iter_affine_map.py | 0
.../{unittest => arith}/test_arith_modular_set.py | 0
.../test_arith_narrow_predicate_expression.py | 0
.../test_arith_rewrite_simplify.py | 0
.../{unittest => arith}/test_arith_simplify.py | 0
.../test_arith_solve_linear_equations.py | 0
.../test_arith_solve_linear_inequality.py | 0
.../test_auto_scheduler_compute_dag.py | 0
.../test_auto_scheduler_cost_model.py | 0
.../test_auto_scheduler_evolutionary_search.py | 0
.../test_auto_scheduler_feature.py | 0
.../test_auto_scheduler_layout_rewrite.py | 0
.../test_auto_scheduler_loop_state.py | 0
.../test_auto_scheduler_measure.py | 0
.../test_auto_scheduler_search_policy.py | 0
.../test_auto_scheduler_search_task.py | 0
.../test_auto_scheduler_sketch_generation.py | 0
.../test_auto_scheduler_task_scheduler.py | 0
.../{unittest => autotvm}/test_autotvm_database.py | 0
.../test_autotvm_dispatch_context.py | 0
.../test_autotvm_droplet_tuner.py | 0
.../{unittest => autotvm}/test_autotvm_feature.py | 0
.../test_autotvm_flop_calculator.py | 0
.../{unittest => autotvm}/test_autotvm_ga_tuner.py | 0
.../test_autotvm_graph_tuner_core.py | 0
.../test_autotvm_graph_tuner_utils.py | 0
.../test_autotvm_index_tuner.py | 0
.../{unittest => autotvm}/test_autotvm_measure.py | 0
.../{unittest => autotvm}/test_autotvm_record.py | 0
.../{unittest => autotvm}/test_autotvm_space.py | 0
.../test_autotvm_xgboost_model.py | 0
.../test_gpu_codegen_allreduce.py} | 0
.../{unittest => codegen}/test_inject_ptx_ldg32.py | 0
.../test_target_codegen_aarch64.py | 0
.../test_target_codegen_arm.py | 0
.../test_target_codegen_blob.py | 0
.../test_target_codegen_bool.py | 0
.../test_target_codegen_c_host.py | 0
.../test_target_codegen_cross_llvm.py | 0
.../test_target_codegen_cuda.py | 0
.../test_target_codegen_device.py | 0
.../test_target_codegen_extern.py | 0
.../test_target_codegen_hexagon.py | 0
.../test_target_codegen_llvm.py | 0
.../test_target_codegen_metal.py | 0
.../test_target_codegen_opencl.py | 0
.../test_target_codegen_rocm.py | 0
.../test_target_codegen_static_init.py | 0
.../test_target_codegen_vm_basic.py | 0
.../test_target_codegen_vulkan.py | 0
.../test_target_codegen_x86.py | 0
.../test_target_texture_codegen_opencl.py | 0
tests/python/conftest.py | 10 +-
tests/python/contrib/test_msc/test_runner.py | 71 +++
.../contrib/test_msc/test_translate_relay.py | 2 +-
tests/python/contrib/test_random.py | 15 +-
tests/python/contrib/test_rocblas.py | 8 +-
tests/python/contrib/test_rpc_proxy.py | 7 +-
tests/python/contrib/test_rpc_server_device.py | 1 -
tests/python/contrib/test_rpc_tracker.py | 13 +-
tests/python/contrib/test_sort.py | 11 +-
tests/python/contrib/test_sparse.py | 12 +-
tests/python/contrib/test_tedd.py | 28 +-
tests/python/disco/test_loader.py | 116 ++++-
tests/python/dlight/test_gpu_matmul.py | 130 ++++-
tests/python/dlight/test_gpu_matmul_tensorize.py | 273 ++++++++++
tests/python/dlight/test_gpu_reduction.py | 68 ++-
tests/python/frontend/pytorch/test_forward.py | 51 ++
tests/python/frontend/tflite/test_forward.py | 2 +-
.../test_container_structural_equal.py | 0
.../{unittest => ir}/test_datatype_nv_fp8.py | 0
tests/python/{unittest => ir}/test_ir_attrs.py | 0
tests/python/{unittest => ir}/test_ir_container.py | 0
tests/python/{unittest => ir}/test_ir_type.py | 0
.../{unittest => ir}/test_node_reflection.py | 0
tests/python/{unittest => ir}/test_object_path.py | 0
.../test_roundtrip_runtime_module.py | 0
.../test_meta_schedule_arg_info.py | 0
.../test_meta_schedule_builder.py | 0
.../test_meta_schedule_byoc_tensorrt.py | 0
.../test_meta_schedule_cost_model.py | 0
.../test_meta_schedule_cpu_dot_product.py | 0
.../test_meta_schedule_database.py | 0
.../test_meta_schedule_feature_extractor.py | 0
...schedule_feature_extractor_per_store_feature.py | 0
.../test_meta_schedule_measure_callback.py | 0
...meta_schedule_mma_m16n8k8_auto_tensorization.py | 0
.../test_meta_schedule_multi_anchor.py | 0
...eta_schedule_mutator_mutate_compute_location.py | 0
.../test_meta_schedule_mutator_mutate_parallel.py | 0
..._meta_schedule_mutator_mutate_thread_binding.py | 0
.../test_meta_schedule_mutator_mutate_tile_size.py | 0
.../test_meta_schedule_mutator_mutate_unroll.py | 0
.../test_meta_schedule_post_order_apply.py | 0
...ule_postproc_disallow_async_strided_mem_copy.py | 0
...meta_schedule_postproc_disallow_dynamic_loop.py | 0
..._schedule_postproc_rewrite_cooperative_fetch.py | 0
.../test_meta_schedule_postproc_rewrite_layout.py | 0
...e_postproc_rewrite_parallel_vectorize_unroll.py | 0
...ta_schedule_postproc_rewrite_reduction_block.py | 0
...est_meta_schedule_postproc_rewrite_tensorize.py | 0
...meta_schedule_postproc_rewrite_unbound_block.py | 0
.../test_meta_schedule_postproc_verify_gpu_code.py | 0
...est_meta_schedule_postproc_verify_vtcm_limit.py | 0
.../test_meta_schedule_profiler.py | 0
.../test_meta_schedule_relay_integration.py | 0
.../test_meta_schedule_relay_tir_compute.py | 0
.../test_meta_schedule_runner.py | 0
...meta_schedule_schedule_cuda_layout_transform.py | 0
...test_meta_schedule_schedule_rule_add_rfactor.py | 0
...eta_schedule_schedule_rule_apply_custom_rule.py | 0
.../test_meta_schedule_schedule_rule_auto_bind.py | 0
...test_meta_schedule_schedule_rule_auto_inline.py | 0
...chedule_schedule_rule_cross_thread_reduction.py | 0
.../test_meta_schedule_schedule_rule_mlt.py | 0
.../test_meta_schedule_schedule_rule_mlt_intrin.py | 0
.../test_meta_schedule_schedule_rule_mlt_tc.py | 0
...dule_schedule_rule_parallel_vectorize_unroll.py | 0
...hedule_schedule_rule_random_compute_location.py | 0
.../test_meta_schedule_search_strategy.py | 0
.../test_meta_schedule_space_cpu.py | 0
.../test_meta_schedule_space_cpu_winograd.py | 0
.../test_meta_schedule_space_cuda.py | 0
.../test_meta_schedule_space_cuda_async.py | 0
...ule_space_cuda_async_multiple_initialization.py | 0
.../test_meta_schedule_space_cuda_winograd.py | 0
.../test_meta_schedule_space_generator.py | 0
.../test_meta_schedule_task_scheduler.py | 0
.../test_meta_schedule_trace_apply.py | 0
.../test_meta_schedule_tune_context.py | 0
.../test_meta_schedule_tune_tir.py | 0
.../test_aot_legalize_packed_call.py | 0
tests/python/{unittest => micro}/test_crt.py | 0
.../test_micro_model_library_format.py | 0
.../{unittest => micro}/test_micro_ms_tuning.py | 0
.../{unittest => micro}/test_micro_project_api.py | 0
.../{unittest => micro}/test_micro_transport.py | 0
tests/python/relax/conftest.py | 56 +-
..._distributed_transform_legalize_redistribute.py | 69 +++
tests/python/relax/test_analysis.py | 30 ++
tests/python/relax/test_codegen_cublas.py | 98 +++-
tests/python/relax/test_dataflow_pattern.py | 1 +
tests/python/relax/test_op_distributed.py | 58 +++
...est_runtime_builtin_paged_attention_kv_cache.py | 5 +
tests/python/relax/test_transform_cse.py | 55 +-
.../relax/test_transform_dead_code_elimination.py | 10 +
tests/python/relax/test_transform_gradient.py | 109 ++++
.../relax/test_transform_gradient_checkpoint.py | 548 +++++++++++---------
.../relax/test_transform_gradient_numeric.py | 37 ++
.../relax/test_transform_lazy_transform_params.py | 174 +++++++
.../relax/test_transform_legalize_ops_ccl.py | 32 +-
.../test_transform_legalize_ops_distributed.py | 64 +++
.../python/relax/test_transform_legalize_ops_nn.py | 289 +++++------
.../relax/test_transform_lift_transform_params.py | 155 ++++++
...st_transform_operator_specific_normalization.py | 377 ++++++++++++++
.../test_transform_rewrite_dataflow_reshape.py | 81 +++
.../test_conv2d_transpose_nchw_texture.py | 325 ++++++++++++
.../relay/opencl_texture/utils/adreno_utils.py | 5 +-
.../{unittest => relay}/test_custom_datatypes.py | 0
.../python/{unittest => relay}/test_link_params.py | 0
.../test_pass_div_to_mul.py} | 0
tests/python/{unittest => relay}/test_roofline.py | 0
.../test_set_input_zero_copy.py | 0
.../test_evaluator_with_preproc.py | 0
.../python/{unittest => runtime}/test_rpc_base.py | 0
.../test_runtime_container.py | 0
.../{unittest => runtime}/test_runtime_dlpack.py | 0
.../{unittest => runtime}/test_runtime_error.py | 0
.../test_runtime_extension.py | 0
.../{unittest => runtime}/test_runtime_graph.py | 0
.../test_runtime_graph_cuda_graph.py | 0
.../test_runtime_graph_debug.py | 0
.../test_runtime_heterogeneous.py | 0
.../{unittest => runtime}/test_runtime_measure.py | 0
.../test_runtime_module_based_interface.py | 0
.../test_runtime_module_export.py | 0
.../test_runtime_module_load.py | 0
.../test_runtime_module_property.py | 0
.../test_runtime_profiling.py | 0
.../{unittest => runtime}/test_runtime_rpc.py | 0
.../{unittest => runtime}/test_runtime_trace.py | 0
.../test_runtime_vm_profiler.py | 0
tests/python/{unittest => target}/test_device.py | 0
.../test_target_parser_mprofile.py | 0
.../{unittest => target}/test_target_target.py | 7 +
tests/python/{unittest => te}/test_te_autodiff.py | 0
.../python/{unittest => te}/test_te_build_lower.py | 0
.../{unittest => te}/test_te_create_primfunc.py | 0
tests/python/{unittest => te}/test_te_group.py | 0
.../{unittest => te}/test_te_hybrid_script.py | 0
tests/python/{unittest => te}/test_te_schedule.py | 0
.../test_te_schedule_bound_inference.py | 0
.../test_te_schedule_bound_inference_tiling.py | 0
.../{unittest => te}/test_te_schedule_graph.py | 0
.../{unittest => te}/test_te_schedule_lstm.py | 0
.../{unittest => te}/test_te_schedule_ops.py | 0
...te_schedule_postproc_rewrite_for_tensor_core.py | 0
.../test_te_schedule_tensor_core.py | 0
.../{unittest => te}/test_te_schedule_tensorize.py | 0
tests/python/{unittest => te}/test_te_tag.py | 0
tests/python/{unittest => te}/test_te_tensor.py | 0
.../{unittest => te}/test_te_tensor_overload.py | 0
.../{unittest => te}/test_te_transform_layout.py | 0
.../{unittest => te}/test_te_verify_compute.py | 0
.../{unittest => testing}/test_filter_untracked.py | 0
.../{unittest => testing}/test_format_si_prefix.py | 0
.../{unittest => testing}/test_gen_requirements.py | 0
tests/python/{unittest => testing}/test_testing.py | 0
.../test_tvm_testing_before_after.py | 0
.../test_tvm_testing_features.py | 0
.../test_type_annotation_checker.py | 0
...test_tir_analysis_calculate_allocated_memory.py | 0
.../test_tir_analysis_calculate_workspace.py | 0
.../test_tir_analysis_detect_buffer_access_lca.py | 0
.../test_tir_analysis_device_constraint_utils.py} | 0
.../test_tir_analysis_estimate_tir_flops.py | 0
.../test_tir_analysis_expr_deep_equal.py | 0
.../test_tir_analysis_get_block_access_region.py | 0
.../test_tir_analysis_identify_memcpy.py | 0
.../test_tir_analysis_oob.py | 0
.../test_tir_analysis_stmt_finding.py | 0
.../test_tir_analysis_usedef.py | 0
.../test_tir_analysis_verify_gpu_code.py | 0
.../test_tir_analysis_verify_memory.py | 0
.../test_tir_analysis_verify_ssa.py | 0
.../test_tir_analysis_verify_well_formed.py | 0
tests/python/{tir => tir-base}/test_debug_info.py | 0
.../{unittest => tir-base}/test_lower_build.py | 0
.../{unittest => tir-base}/test_slice_tir.py | 0
.../python/{unittest => tir-base}/test_tir_base.py | 0
.../test_tir_block_dependence_info.py | 0
.../{unittest => tir-base}/test_tir_buffer.py | 0
.../{unittest => tir-base}/test_tir_constructor.py | 0
.../{unittest => tir-base}/test_tir_data_layout.py | 0
.../{unittest => tir-base}/test_tir_host_func.py | 0
.../{unittest => tir-base}/test_tir_imm_values.py | 0
.../{unittest => tir-base}/test_tir_index_map.py | 0
.../{unittest => tir-base}/test_tir_intrin.py | 0
.../{unittest => tir-base}/test_tir_ir_builder.py | 0
.../{unittest => tir-base}/test_tir_nodes.py | 0
.../{unittest => tir-base}/test_tir_op_types.py | 0
.../python/{unittest => tir-base}/test_tir_ops.py | 0
.../test_tir_ptx_cp_async.py | 0
.../test_tir_ptx_ldmatrix.py | 0
.../{unittest => tir-base}/test_tir_ptx_mma.py | 0
.../{unittest => tir-base}/test_tir_ptx_mma_sp.py | 0
.../{unittest => tir-base}/test_tir_renew_defs.py | 0
.../{unittest => tir-base}/test_tir_specialize.py | 0
.../test_tir_stmt_functor_ir_transform.py | 0
.../test_tir_stmt_functor_substitute.py | 0
.../test_tir_structural_equal_hash.py | 0
.../test_tir_te_extern_primfunc.py | 0
.../test_tir_texture_scope.py | 0
.../test_tir_unsafe_hide_buffer_access.py | 0
.../test_tir_schedule_analysis.py | 2 +-
.../test_tir_schedule_block_scope.py | 0
.../test_tir_schedule_blockize.py | 0
.../test_tir_schedule_cache_index.py | 0
.../test_tir_schedule_cache_read_write.py | 0
.../test_tir_schedule_compute_at.py | 0
.../test_tir_schedule_compute_inline.py | 0
.../test_tir_schedule_decompose_padding.py | 0
.../test_tir_schedule_error.py | 0
.../test_tir_schedule_for_kind.py | 0
.../test_tir_schedule_instruction.py | 0
.../test_tir_schedule_merge.py | 0
.../test_tir_schedule_pad_einsum.py | 0
.../test_tir_schedule_read_write_at.py | 0
.../test_tir_schedule_reduction.py | 0
.../test_tir_schedule_reindex.py | 0
.../test_tir_schedule_reorder.py | 0
.../test_tir_schedule_reorder_block_iter_var.py} | 0
.../test_tir_schedule_rfactor.py | 0
.../test_tir_schedule_rolling_buffer.py | 0
.../test_tir_schedule_sampling.py | 0
.../test_tir_schedule_set_axis_separator.py | 0
.../test_tir_schedule_set_dtype.py | 0
.../test_tir_schedule_set_scope.py | 0
.../test_tir_schedule_split_fuse.py | 0
.../test_tir_schedule_state.py | 2 +-
.../test_tir_schedule_state_cached_flags.py | 0
.../test_tir_schedule_storage_align.py | 0
.../test_tir_schedule_tensorize.py | 0
...tir_schedule_tensorize_ldmatrix_mma_numeric.py} | 56 +-
.../test_tir_schedule_tensorize_mfma_numeric.py} | 0
.../test_tir_schedule_trace.py | 0
.../test_tir_schedule_transform.py | 0
.../test_tir_schedule_transform_layout.py | 0
.../test_tir_schedule_utilities.py | 0
.../test_tir_transform_annotate_device_regions.py | 0
.../test_tir_transform_bf16_legalize.py | 0
.../test_tir_transform_combine_context_call.py | 0
.../test_tir_transform_common_subexpr_elim.py | 0
.../test_tir_transform_compact_buffer_region.py | 0
.../test_tir_transform_convert_blocks_to_opaque.py | 0
.../test_tir_transform_convert_for_loops_serial.py | 0
.../test_tir_transform_convert_ssa.py | 0
.../test_tir_transform_coproc_sync.py | 0
.../test_tir_transform_decorate_device_scope.py | 0
.../test_tir_transform_device_kernel_launch.py | 0
.../test_tir_transform_extract_constants.py | 0
.../test_tir_transform_flatten_buffer.py | 0
...test_tir_transform_force_narrow_index_to_i32.py | 0
.../test_tir_transform_fp8_legalize.py | 0
.../test_tir_transform_helpers.py | 0
.../test_tir_transform_hoist_expression.py | 0
.../test_tir_transform_hoist_if.py | 0
.../test_tir_transform_inject_copy_intrin.py | 0
.../test_tir_transform_inject_double_buffer.py | 0
.../test_tir_transform_inject_permuted_layout.py | 351 +++++++++++++
.../test_tir_transform_inject_ptx_async_copy.py | 0
.../test_tir_transform_inject_rolling_buffer.py | 0
.../test_tir_transform_inject_software_pipeline.py | 8 +-
.../test_tir_transform_inject_virtual_thread.py | 0
...test_tir_transform_instrument_bound_checkers.py | 0
.../test_tir_transform_lift_attr_scope.py | 0
.../test_tir_transform_lift_thread_binding.py | 0
.../test_tir_transform_loop_partition.py | 0
...t_tir_transform_lower_cross_thread_reduction.py | 0
...r_transform_lower_device_storage_access_info.py | 0
.../test_tir_transform_lower_init_block.py | 0
.../test_tir_transform_lower_intrin.py | 0
.../test_tir_transform_lower_match_buffer.py} | 0
.../test_tir_transform_lower_opaque_block.py | 0
.../test_tir_transform_lower_thread_all_reduce.py | 0
.../test_tir_transform_lower_tvm_builtin.py | 0
.../test_tir_transform_lower_warp_memory.py | 0
.../test_tir_transform_make_packed_api.py | 0
.../test_tir_transform_make_unpacked_api.py | 0
...transform_manifest_shared_memory_local_stage.py | 0
...test_tir_transform_memhammer_lower_auto_copy.py | 0
...form_merge_dynamic_shared_memory_allocations.py | 0
.../test_tir_transform_narrow_datatype.py | 0
...sform_plan_update_buffer_allocation_location.py | 0
...est_tir_transform_pointer_value_type_rewrite.py | 0
.../test_tir_transform_prim_func_pass.py | 0
.../test_tir_transform_profiling_instr.py | 0
...ansform_reduce_branching_through_overcompute.py | 0
.../test_tir_transform_remove_assume.py | 0
.../test_tir_transform_remove_no_op.py | 0
.../test_tir_transform_remove_undef.py | 0
...transform_remove_weight_layout_rewrite_block.py | 0
...test_tir_transform_renormalize_split_pattern.py | 0
.../test_tir_transform_rewrite_unsafe_select.py | 0
.../test_tir_transform_simplify.py | 0
.../test_tir_transform_split_host_device.py | 0
.../test_tir_transform_storage_flatten.py | 0
.../test_tir_transform_storage_rewrite.py | 0
.../test_tir_transform_thread_sync.py | 0
.../test_tir_transform_unify_thread_binding.py | 0
.../test_tir_transform_unroll_loop.py | 0
.../test_tir_transform_vectorize.py | 0
.../test_transform_default_gpu_schedule.py | 0
.../{unittest => tir-usmp}/test_tir_usmp_algo.py | 0
.../test_tir_usmp_algo_hill_climb.py | 0
.../test_tir_usmp_analysis_extract_bufferinfo.py | 0
...ransform_convert_pool_allocations_to_offsets.py | 0
.../test_tir_usmp_transform_create_io_allocates.py | 0
.../{unittest => tir-usmp}/test_tir_usmp_utils.py | 0
tests/python/topi/{python => }/common.py | 0
tests/python/topi/{python => }/test_fifo_buffer.py | 0
.../python/topi/{python => }/test_topi_argwhere.py | 0
tests/python/topi/{python => }/test_topi_basic.py | 0
.../topi/{python => }/test_topi_batch_matmul.py | 0
.../test_topi_batch_matmul_tensorcore.py | 0
.../topi/{python => }/test_topi_batch_norm.py | 0
.../{python => }/test_topi_batch_to_space_nd.py | 0
.../{python => }/test_topi_bitserial_conv2d.py | 0
.../test_topi_bitserial_conv2d_rasp.py | 0
.../topi/{python => }/test_topi_bitserial_dense.py | 0
tests/python/topi/{python => }/test_topi_bnn.py | 0
.../topi/{python => }/test_topi_broadcast.py | 0
tests/python/topi/{python => }/test_topi_clip.py | 0
tests/python/topi/{python => }/test_topi_conv1d.py | 0
.../{python => }/test_topi_conv1d_transpose_ncw.py | 0
.../topi/{python => }/test_topi_conv2d_NCHWc.py | 0
.../topi/{python => }/test_topi_conv2d_hwcn.py | 0
.../test_topi_conv2d_hwnc_tensorcore.py | 0
.../topi/{python => }/test_topi_conv2d_int8.py | 0
.../topi/{python => }/test_topi_conv2d_nchw.py | 0
.../topi/{python => }/test_topi_conv2d_nhwc.py | 0
.../test_topi_conv2d_nhwc_pack_int8.py | 0
.../test_topi_conv2d_nhwc_tensorcore.py | 0
.../{python => }/test_topi_conv2d_nhwc_winograd.py | 0
.../test_topi_conv2d_tensordot_opts.py | 0
.../test_topi_conv2d_transpose_nchw.py | 0
.../topi/{python => }/test_topi_conv2d_winograd.py | 0
.../topi/{python => }/test_topi_conv3d_ncdhw.py | 0
.../topi/{python => }/test_topi_conv3d_ndhwc.py | 0
.../test_topi_conv3d_ndhwc_tensorcore.py | 0
.../test_topi_conv3d_transpose_ncdhw.py | 0
.../topi/{python => }/test_topi_conv3d_winograd.py | 0
.../topi/{python => }/test_topi_correlation.py | 0
.../{python => }/test_topi_deformable_conv2d.py | 0
tests/python/topi/{python => }/test_topi_dense.py | 0
.../{python => }/test_topi_dense_tensorcore.py | 0
.../topi/{python => }/test_topi_depth_to_space.py | 0
.../{python => }/test_topi_depthwise_conv2d.py | 0
.../test_topi_depthwise_conv2d_back_input.py | 0
.../test_topi_depthwise_conv2d_back_weight.py | 0
tests/python/topi/{python => }/test_topi_dft.py | 0
tests/python/topi/{python => }/test_topi_dilate.py | 0
tests/python/topi/{python => }/test_topi_einsum.py | 0
.../topi/{python => }/test_topi_group_conv2d.py | 0
.../test_topi_group_conv2d_NCHWc_int8.py | 0
.../test_topi_group_conv2d_transpose.py | 0
.../topi/{python => }/test_topi_group_norm.py | 0
tests/python/topi/{python => }/test_topi_image.py | 0
.../topi/{python => }/test_topi_instance_norm.py | 0
.../topi/{python => }/test_topi_layer_norm.py | 0
tests/python/topi/{python => }/test_topi_loss.py | 0
tests/python/topi/{python => }/test_topi_lrn.py | 0
tests/python/topi/{python => }/test_topi_lstm.py | 0
tests/python/topi/{python => }/test_topi_math.py | 0
tests/python/topi/{python => }/test_topi_matmul.py | 61 ++-
.../python/topi/{python => }/test_topi_pooling.py | 0
tests/python/topi/{python => }/test_topi_prng.py | 0
tests/python/topi/{python => }/test_topi_qnn.py | 0
tests/python/topi/{python => }/test_topi_reduce.py | 0
tests/python/topi/{python => }/test_topi_relu.py | 0
tests/python/topi/{python => }/test_topi_reorg.py | 0
.../python/topi/{python => }/test_topi_rms_norm.py | 14 +-
tests/python/topi/{python => }/test_topi_scan.py | 0
.../python/topi/{python => }/test_topi_scatter.py | 0
.../topi/{python => }/test_topi_searchsorted.py | 0
.../python/topi/{python => }/test_topi_softmax.py | 0
tests/python/topi/{python => }/test_topi_sort.py | 0
.../{python => }/test_topi_space_to_batch_nd.py | 0
.../topi/{python => }/test_topi_space_to_depth.py | 0
tests/python/topi/{python => }/test_topi_sparse.py | 0
tests/python/topi/{python => }/test_topi_tensor.py | 0
.../topi/{python => }/test_topi_transform.py | 0
tests/python/topi/{python => }/test_topi_unique.py | 0
.../topi/{python => }/test_topi_upsampling.py | 0
tests/python/topi/{python => }/test_topi_util.py | 0
tests/python/topi/{python => }/test_topi_vision.py | 0
.../test_tvmscript_complete.py | 0
.../test_tvmscript_error_report.py | 0
.../test_tvmscript_ir_builder_base.py | 0
.../test_tvmscript_ir_builder_irmodule.py | 0
.../test_tvmscript_ir_builder_tir.py | 12 +-
.../test_tvmscript_meta_programming.py | 0
.../{unittest => tvmscript}/test_tvmscript_ops.py | 0
.../test_tvmscript_parser_evaluator.py | 0
.../test_tvmscript_parser_ir.py | 0
.../test_tvmscript_parser_source.py | 0
.../test_tvmscript_parser_tir.py | 0
.../test_tvmscript_printer_annotation.py | 0
.../test_tvmscript_printer_doc.py | 0
.../test_tvmscript_printer_highlight.py | 0
.../test_tvmscript_printer_ir.py | 0
.../test_tvmscript_printer_metadata.py | 0
.../test_tvmscript_printer_python_doc_printer.py | 0
.../test_tvmscript_printer_structural_equal.py | 0
.../test_tvmscript_printer_tir.py | 0
.../test_tvmscript_printer_underlining.py | 0
.../test_tvmscript_regression.py | 0
.../test_tvmscript_roundtrip.py | 26 +
.../test_tvmscript_syntax_sugar.py | 0
.../{unittest => tvmscript}/test_tvmscript_type.py | 0
tests/scripts/task_mypy.sh | 2 +-
tests/scripts/task_python_docs.sh | 4 +-
tests/scripts/task_python_unittest.sh | 28 +-
tests/scripts/task_python_unittest_gpuonly.sh | 6 +-
vta/runtime/runtime.cc | 12 +-
web/emcc/wasm_runtime.cc | 2 +-
web/src/webgpu.ts | 22 +-
602 files changed, 7858 insertions(+), 1672 deletions(-)
copy python/tvm/{relax/distributed/transform =>
contrib/msc/framework/tensorflow/runtime}/__init__.py (89%)
create mode 100644
python/tvm/contrib/msc/framework/tensorflow/runtime/runner.py
copy python/tvm/{relax/distributed/transform =>
contrib/msc/framework/tensorrt/runtime}/__init__.py (89%)
copy tests/python/unittest/test_tvmscript_ir_builder_base.py =>
python/tvm/contrib/msc/framework/tensorrt/runtime/runner.py (50%)
copy python/tvm/{relax/distributed/transform =>
contrib/msc/framework/torch/runtime}/__init__.py (89%)
create mode 100644 python/tvm/contrib/msc/framework/torch/runtime/runner.py
create mode 100644 python/tvm/relax/transform/legalize_ops/distributed.py
copy python/tvm/{relax/distributed/transform/__init__.py =>
script/_ffi_api.py} (82%)
create mode 100644 python/tvm/topi/adreno/conv2d_transpose_alter_op.py
create mode 100644 python/tvm/topi/adreno/conv2d_transpose_nchw.py
create mode 100644 src/contrib/msc/core/transform/set_byoc_attrs.cc
create mode 100644 src/relax/distributed/transform/legalize_redistribute.cc
copy src/relax/{op/ccl/ccl.h => testing/transform.cc} (56%)
create mode 100644 src/relax/transform/gradient_simplifier.cc
copy src/relax/{op/ccl/ccl.h => transform/gradient_simplifier.h} (58%)
rename tests/python/{unittest => arith}/test_arith_canonical_simplify.py (100%)
rename tests/python/{unittest => arith}/test_arith_const_int_bound.py (100%)
rename tests/python/{unittest => arith}/test_arith_deduce_bound.py (100%)
rename tests/python/{unittest => arith}/test_arith_detect_clip_bound.py (100%)
rename tests/python/{unittest => arith}/test_arith_detect_cse.py (100%)
rename tests/python/{unittest => arith}/test_arith_detect_linear_equation.py
(100%)
rename tests/python/{unittest => arith}/test_arith_domain_touched.py (100%)
rename tests/python/{unittest => arith}/test_arith_intset.py (100%)
rename tests/python/{unittest => arith}/test_arith_iter_affine_map.py (100%)
rename tests/python/{unittest => arith}/test_arith_modular_set.py (100%)
rename tests/python/{unittest =>
arith}/test_arith_narrow_predicate_expression.py (100%)
rename tests/python/{unittest => arith}/test_arith_rewrite_simplify.py (100%)
rename tests/python/{unittest => arith}/test_arith_simplify.py (100%)
rename tests/python/{unittest => arith}/test_arith_solve_linear_equations.py
(100%)
rename tests/python/{unittest => arith}/test_arith_solve_linear_inequality.py
(100%)
rename tests/python/{unittest =>
auto_scheduler}/test_auto_scheduler_compute_dag.py (100%)
rename tests/python/{unittest =>
auto_scheduler}/test_auto_scheduler_cost_model.py (100%)
rename tests/python/{unittest =>
auto_scheduler}/test_auto_scheduler_evolutionary_search.py (100%)
rename tests/python/{unittest =>
auto_scheduler}/test_auto_scheduler_feature.py (100%)
rename tests/python/{unittest =>
auto_scheduler}/test_auto_scheduler_layout_rewrite.py (100%)
rename tests/python/{unittest =>
auto_scheduler}/test_auto_scheduler_loop_state.py (100%)
rename tests/python/{unittest =>
auto_scheduler}/test_auto_scheduler_measure.py (100%)
rename tests/python/{unittest =>
auto_scheduler}/test_auto_scheduler_search_policy.py (100%)
rename tests/python/{unittest =>
auto_scheduler}/test_auto_scheduler_search_task.py (100%)
rename tests/python/{unittest =>
auto_scheduler}/test_auto_scheduler_sketch_generation.py (100%)
rename tests/python/{unittest =>
auto_scheduler}/test_auto_scheduler_task_scheduler.py (100%)
rename tests/python/{unittest => autotvm}/test_autotvm_database.py (100%)
rename tests/python/{unittest => autotvm}/test_autotvm_dispatch_context.py
(100%)
rename tests/python/{unittest => autotvm}/test_autotvm_droplet_tuner.py (100%)
rename tests/python/{unittest => autotvm}/test_autotvm_feature.py (100%)
rename tests/python/{unittest => autotvm}/test_autotvm_flop_calculator.py
(100%)
rename tests/python/{unittest => autotvm}/test_autotvm_ga_tuner.py (100%)
rename tests/python/{unittest => autotvm}/test_autotvm_graph_tuner_core.py
(100%)
rename tests/python/{unittest => autotvm}/test_autotvm_graph_tuner_utils.py
(100%)
rename tests/python/{unittest => autotvm}/test_autotvm_index_tuner.py (100%)
rename tests/python/{unittest => autotvm}/test_autotvm_measure.py (100%)
rename tests/python/{unittest => autotvm}/test_autotvm_record.py (100%)
rename tests/python/{unittest => autotvm}/test_autotvm_space.py (100%)
rename tests/python/{unittest => autotvm}/test_autotvm_xgboost_model.py (100%)
rename tests/python/{unittest/test_allreduce.py =>
codegen/test_gpu_codegen_allreduce.py} (100%)
rename tests/python/{unittest => codegen}/test_inject_ptx_ldg32.py (100%)
rename tests/python/{unittest => codegen}/test_target_codegen_aarch64.py (100%)
rename tests/python/{unittest => codegen}/test_target_codegen_arm.py (100%)
rename tests/python/{unittest => codegen}/test_target_codegen_blob.py (100%)
rename tests/python/{unittest => codegen}/test_target_codegen_bool.py (100%)
rename tests/python/{unittest => codegen}/test_target_codegen_c_host.py (100%)
rename tests/python/{unittest => codegen}/test_target_codegen_cross_llvm.py
(100%)
rename tests/python/{unittest => codegen}/test_target_codegen_cuda.py (100%)
rename tests/python/{unittest => codegen}/test_target_codegen_device.py (100%)
rename tests/python/{unittest => codegen}/test_target_codegen_extern.py (100%)
rename tests/python/{unittest => codegen}/test_target_codegen_hexagon.py (100%)
rename tests/python/{unittest => codegen}/test_target_codegen_llvm.py (100%)
rename tests/python/{unittest => codegen}/test_target_codegen_metal.py (100%)
rename tests/python/{unittest => codegen}/test_target_codegen_opencl.py (100%)
rename tests/python/{unittest => codegen}/test_target_codegen_rocm.py (100%)
rename tests/python/{unittest => codegen}/test_target_codegen_static_init.py
(100%)
rename tests/python/{unittest => codegen}/test_target_codegen_vm_basic.py
(100%)
rename tests/python/{unittest => codegen}/test_target_codegen_vulkan.py (100%)
rename tests/python/{unittest => codegen}/test_target_codegen_x86.py (100%)
rename tests/python/{unittest =>
codegen}/test_target_texture_codegen_opencl.py (100%)
rename tests/python/{unittest => ir}/test_container_structural_equal.py (100%)
rename tests/python/{unittest => ir}/test_datatype_nv_fp8.py (100%)
rename tests/python/{unittest => ir}/test_ir_attrs.py (100%)
rename tests/python/{unittest => ir}/test_ir_container.py (100%)
rename tests/python/{unittest => ir}/test_ir_type.py (100%)
rename tests/python/{unittest => ir}/test_node_reflection.py (100%)
rename tests/python/{unittest => ir}/test_object_path.py (100%)
rename tests/python/{unittest => ir}/test_roundtrip_runtime_module.py (100%)
rename tests/python/{unittest => meta_schedule}/test_meta_schedule_arg_info.py
(100%)
rename tests/python/{unittest => meta_schedule}/test_meta_schedule_builder.py
(100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_byoc_tensorrt.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_cost_model.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_cpu_dot_product.py (100%)
rename tests/python/{unittest => meta_schedule}/test_meta_schedule_database.py
(100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_feature_extractor.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_feature_extractor_per_store_feature.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_measure_callback.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_mma_m16n8k8_auto_tensorization.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_multi_anchor.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_mutator_mutate_compute_location.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_mutator_mutate_parallel.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_mutator_mutate_thread_binding.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_mutator_mutate_tile_size.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_mutator_mutate_unroll.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_post_order_apply.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_postproc_disallow_async_strided_mem_copy.py
(100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_postproc_disallow_dynamic_loop.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_postproc_rewrite_cooperative_fetch.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_postproc_rewrite_layout.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_postproc_rewrite_parallel_vectorize_unroll.py
(100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_postproc_rewrite_reduction_block.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_postproc_rewrite_tensorize.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_postproc_rewrite_unbound_block.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_postproc_verify_gpu_code.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_postproc_verify_vtcm_limit.py (100%)
rename tests/python/{unittest => meta_schedule}/test_meta_schedule_profiler.py
(100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_relay_integration.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_relay_tir_compute.py (100%)
rename tests/python/{unittest => meta_schedule}/test_meta_schedule_runner.py
(100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_schedule_cuda_layout_transform.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_schedule_rule_add_rfactor.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_schedule_rule_apply_custom_rule.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_schedule_rule_auto_bind.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_schedule_rule_auto_inline.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_schedule_rule_cross_thread_reduction.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_schedule_rule_mlt.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_schedule_rule_mlt_intrin.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_schedule_rule_mlt_tc.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_schedule_rule_parallel_vectorize_unroll.py
(100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_schedule_rule_random_compute_location.py
(100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_search_strategy.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_space_cpu.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_space_cpu_winograd.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_space_cuda.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_space_cuda_async.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_space_cuda_async_multiple_initialization.py
(100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_space_cuda_winograd.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_space_generator.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_task_scheduler.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_trace_apply.py (100%)
rename tests/python/{unittest =>
meta_schedule}/test_meta_schedule_tune_context.py (100%)
rename tests/python/{unittest => meta_schedule}/test_meta_schedule_tune_tir.py
(100%)
rename tests/python/{unittest => micro}/test_aot_legalize_packed_call.py (100%)
rename tests/python/{unittest => micro}/test_crt.py (100%)
rename tests/python/{unittest => micro}/test_micro_model_library_format.py
(100%)
rename tests/python/{unittest => micro}/test_micro_ms_tuning.py (100%)
rename tests/python/{unittest => micro}/test_micro_project_api.py (100%)
rename tests/python/{unittest => micro}/test_micro_transport.py (100%)
create mode 100644
tests/python/relax/distributed/test_distributed_transform_legalize_redistribute.py
create mode 100644 tests/python/relax/test_op_distributed.py
create mode 100644
tests/python/relax/test_transform_legalize_ops_distributed.py
create mode 100644
tests/python/relax/test_transform_operator_specific_normalization.py
create mode 100644
tests/python/relay/opencl_texture/test_conv2d_transpose_nchw_texture.py
rename tests/python/{unittest => relay}/test_custom_datatypes.py (100%)
rename tests/python/{unittest => relay}/test_link_params.py (100%)
rename tests/python/{unittest/test_div_to_mul.py =>
relay/test_pass_div_to_mul.py} (100%)
rename tests/python/{unittest => relay}/test_roofline.py (100%)
rename tests/python/{unittest => relay}/test_set_input_zero_copy.py (100%)
rename tests/python/{unittest => runtime}/test_evaluator_with_preproc.py (100%)
rename tests/python/{unittest => runtime}/test_rpc_base.py (100%)
rename tests/python/{unittest => runtime}/test_runtime_container.py (100%)
rename tests/python/{unittest => runtime}/test_runtime_dlpack.py (100%)
rename tests/python/{unittest => runtime}/test_runtime_error.py (100%)
rename tests/python/{unittest => runtime}/test_runtime_extension.py (100%)
rename tests/python/{unittest => runtime}/test_runtime_graph.py (100%)
rename tests/python/{unittest => runtime}/test_runtime_graph_cuda_graph.py
(100%)
rename tests/python/{unittest => runtime}/test_runtime_graph_debug.py (100%)
rename tests/python/{unittest => runtime}/test_runtime_heterogeneous.py (100%)
rename tests/python/{unittest => runtime}/test_runtime_measure.py (100%)
rename tests/python/{unittest =>
runtime}/test_runtime_module_based_interface.py (100%)
rename tests/python/{unittest => runtime}/test_runtime_module_export.py (100%)
rename tests/python/{unittest => runtime}/test_runtime_module_load.py (100%)
rename tests/python/{unittest => runtime}/test_runtime_module_property.py
(100%)
rename tests/python/{unittest => runtime}/test_runtime_profiling.py (100%)
rename tests/python/{unittest => runtime}/test_runtime_rpc.py (100%)
rename tests/python/{unittest => runtime}/test_runtime_trace.py (100%)
rename tests/python/{unittest => runtime}/test_runtime_vm_profiler.py (100%)
rename tests/python/{unittest => target}/test_device.py (100%)
rename tests/python/{unittest => target}/test_target_parser_mprofile.py (100%)
rename tests/python/{unittest => target}/test_target_target.py (98%)
rename tests/python/{unittest => te}/test_te_autodiff.py (100%)
rename tests/python/{unittest => te}/test_te_build_lower.py (100%)
rename tests/python/{unittest => te}/test_te_create_primfunc.py (100%)
rename tests/python/{unittest => te}/test_te_group.py (100%)
rename tests/python/{unittest => te}/test_te_hybrid_script.py (100%)
rename tests/python/{unittest => te}/test_te_schedule.py (100%)
rename tests/python/{unittest => te}/test_te_schedule_bound_inference.py (100%)
rename tests/python/{unittest =>
te}/test_te_schedule_bound_inference_tiling.py (100%)
rename tests/python/{unittest => te}/test_te_schedule_graph.py (100%)
rename tests/python/{unittest => te}/test_te_schedule_lstm.py (100%)
rename tests/python/{unittest => te}/test_te_schedule_ops.py (100%)
rename tests/python/{unittest =>
te}/test_te_schedule_postproc_rewrite_for_tensor_core.py (100%)
rename tests/python/{unittest => te}/test_te_schedule_tensor_core.py (100%)
rename tests/python/{unittest => te}/test_te_schedule_tensorize.py (100%)
rename tests/python/{unittest => te}/test_te_tag.py (100%)
rename tests/python/{unittest => te}/test_te_tensor.py (100%)
rename tests/python/{unittest => te}/test_te_tensor_overload.py (100%)
rename tests/python/{unittest => te}/test_te_transform_layout.py (100%)
rename tests/python/{unittest => te}/test_te_verify_compute.py (100%)
rename tests/python/{unittest => testing}/test_filter_untracked.py (100%)
rename tests/python/{unittest => testing}/test_format_si_prefix.py (100%)
rename tests/python/{unittest => testing}/test_gen_requirements.py (100%)
rename tests/python/{unittest => testing}/test_testing.py (100%)
rename tests/python/{unittest => testing}/test_tvm_testing_before_after.py
(100%)
rename tests/python/{unittest => testing}/test_tvm_testing_features.py (100%)
rename tests/python/{unittest => testing}/test_type_annotation_checker.py
(100%)
rename tests/python/{unittest =>
tir-analysis}/test_tir_analysis_calculate_allocated_memory.py (100%)
rename tests/python/{unittest =>
tir-analysis}/test_tir_analysis_calculate_workspace.py (100%)
rename tests/python/{unittest =>
tir-analysis}/test_tir_analysis_detect_buffer_access_lca.py (100%)
rename tests/python/{tir/analysis/test_device_constraint_utils.py =>
tir-analysis/test_tir_analysis_device_constraint_utils.py} (100%)
rename tests/python/{unittest =>
tir-analysis}/test_tir_analysis_estimate_tir_flops.py (100%)
rename tests/python/{unittest =>
tir-analysis}/test_tir_analysis_expr_deep_equal.py (100%)
rename tests/python/{unittest =>
tir-analysis}/test_tir_analysis_get_block_access_region.py (100%)
rename tests/python/{unittest =>
tir-analysis}/test_tir_analysis_identify_memcpy.py (100%)
rename tests/python/{unittest => tir-analysis}/test_tir_analysis_oob.py (100%)
rename tests/python/{unittest =>
tir-analysis}/test_tir_analysis_stmt_finding.py (100%)
rename tests/python/{unittest => tir-analysis}/test_tir_analysis_usedef.py
(100%)
rename tests/python/{unittest =>
tir-analysis}/test_tir_analysis_verify_gpu_code.py (100%)
rename tests/python/{unittest =>
tir-analysis}/test_tir_analysis_verify_memory.py (100%)
rename tests/python/{unittest => tir-analysis}/test_tir_analysis_verify_ssa.py
(100%)
rename tests/python/{unittest =>
tir-analysis}/test_tir_analysis_verify_well_formed.py (100%)
rename tests/python/{tir => tir-base}/test_debug_info.py (100%)
rename tests/python/{unittest => tir-base}/test_lower_build.py (100%)
rename tests/python/{unittest => tir-base}/test_slice_tir.py (100%)
rename tests/python/{unittest => tir-base}/test_tir_base.py (100%)
rename tests/python/{unittest => tir-base}/test_tir_block_dependence_info.py
(100%)
rename tests/python/{unittest => tir-base}/test_tir_buffer.py (100%)
rename tests/python/{unittest => tir-base}/test_tir_constructor.py (100%)
rename tests/python/{unittest => tir-base}/test_tir_data_layout.py (100%)
rename tests/python/{unittest => tir-base}/test_tir_host_func.py (100%)
rename tests/python/{unittest => tir-base}/test_tir_imm_values.py (100%)
rename tests/python/{unittest => tir-base}/test_tir_index_map.py (100%)
rename tests/python/{unittest => tir-base}/test_tir_intrin.py (100%)
rename tests/python/{unittest => tir-base}/test_tir_ir_builder.py (100%)
rename tests/python/{unittest => tir-base}/test_tir_nodes.py (100%)
rename tests/python/{unittest => tir-base}/test_tir_op_types.py (100%)
rename tests/python/{unittest => tir-base}/test_tir_ops.py (100%)
rename tests/python/{unittest => tir-base}/test_tir_ptx_cp_async.py (100%)
rename tests/python/{unittest => tir-base}/test_tir_ptx_ldmatrix.py (100%)
rename tests/python/{unittest => tir-base}/test_tir_ptx_mma.py (100%)
rename tests/python/{unittest => tir-base}/test_tir_ptx_mma_sp.py (100%)
rename tests/python/{unittest => tir-base}/test_tir_renew_defs.py (100%)
rename tests/python/{unittest => tir-base}/test_tir_specialize.py (100%)
rename tests/python/{unittest =>
tir-base}/test_tir_stmt_functor_ir_transform.py (100%)
rename tests/python/{unittest => tir-base}/test_tir_stmt_functor_substitute.py
(100%)
rename tests/python/{unittest => tir-base}/test_tir_structural_equal_hash.py
(100%)
rename tests/python/{unittest => tir-base}/test_tir_te_extern_primfunc.py
(100%)
rename tests/python/{unittest => tir-base}/test_tir_texture_scope.py (100%)
rename tests/python/{unittest =>
tir-base}/test_tir_unsafe_hide_buffer_access.py (100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_analysis.py
(99%)
rename tests/python/{unittest =>
tir-schedule}/test_tir_schedule_block_scope.py (100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_blockize.py
(100%)
rename tests/python/{unittest =>
tir-schedule}/test_tir_schedule_cache_index.py (100%)
rename tests/python/{unittest =>
tir-schedule}/test_tir_schedule_cache_read_write.py (100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_compute_at.py
(100%)
rename tests/python/{unittest =>
tir-schedule}/test_tir_schedule_compute_inline.py (100%)
rename tests/python/{unittest =>
tir-schedule}/test_tir_schedule_decompose_padding.py (100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_error.py
(100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_for_kind.py
(100%)
rename tests/python/{unittest =>
tir-schedule}/test_tir_schedule_instruction.py (100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_merge.py
(100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_pad_einsum.py
(100%)
rename tests/python/{unittest =>
tir-schedule}/test_tir_schedule_read_write_at.py (100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_reduction.py
(100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_reindex.py
(100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_reorder.py
(100%)
rename tests/python/{unittest/test_tir_reorder_block_iter_var.py =>
tir-schedule/test_tir_schedule_reorder_block_iter_var.py} (100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_rfactor.py
(100%)
rename tests/python/{unittest =>
tir-schedule}/test_tir_schedule_rolling_buffer.py (100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_sampling.py
(100%)
rename tests/python/{unittest =>
tir-schedule}/test_tir_schedule_set_axis_separator.py (100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_set_dtype.py
(100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_set_scope.py
(100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_split_fuse.py
(100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_state.py (99%)
rename tests/python/{unittest =>
tir-schedule}/test_tir_schedule_state_cached_flags.py (100%)
rename tests/python/{unittest =>
tir-schedule}/test_tir_schedule_storage_align.py (100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_tensorize.py
(100%)
rename tests/python/{unittest/test_tir_schedule_tensorize_ldmatrix_mma.py =>
tir-schedule/test_tir_schedule_tensorize_ldmatrix_mma_numeric.py} (88%)
rename tests/python/{unittest/test_tir_schedule_tensorize_mfma.py =>
tir-schedule/test_tir_schedule_tensorize_mfma_numeric.py} (100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_trace.py
(100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_transform.py
(100%)
rename tests/python/{unittest =>
tir-schedule}/test_tir_schedule_transform_layout.py (100%)
rename tests/python/{unittest => tir-schedule}/test_tir_schedule_utilities.py
(100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_annotate_device_regions.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_bf16_legalize.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_combine_context_call.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_common_subexpr_elim.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_compact_buffer_region.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_convert_blocks_to_opaque.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_convert_for_loops_serial.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_convert_ssa.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_coproc_sync.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_decorate_device_scope.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_device_kernel_launch.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_extract_constants.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_flatten_buffer.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_force_narrow_index_to_i32.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_fp8_legalize.py (100%)
rename tests/python/{unittest => tir-transform}/test_tir_transform_helpers.py
(100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_hoist_expression.py (100%)
rename tests/python/{unittest => tir-transform}/test_tir_transform_hoist_if.py
(100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_inject_copy_intrin.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_inject_double_buffer.py (100%)
create mode 100644
tests/python/tir-transform/test_tir_transform_inject_permuted_layout.py
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_inject_ptx_async_copy.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_inject_rolling_buffer.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_inject_software_pipeline.py (99%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_inject_virtual_thread.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_instrument_bound_checkers.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_lift_attr_scope.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_lift_thread_binding.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_loop_partition.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_lower_cross_thread_reduction.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_lower_device_storage_access_info.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_lower_init_block.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_lower_intrin.py (100%)
rename tests/python/{unittest/test_tir_lower_match_buffer.py =>
tir-transform/test_tir_transform_lower_match_buffer.py} (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_lower_opaque_block.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_lower_thread_all_reduce.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_lower_tvm_builtin.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_lower_warp_memory.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_make_packed_api.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_make_unpacked_api.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_manifest_shared_memory_local_stage.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_memhammer_lower_auto_copy.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_merge_dynamic_shared_memory_allocations.py
(100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_narrow_datatype.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_plan_update_buffer_allocation_location.py
(100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_pointer_value_type_rewrite.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_prim_func_pass.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_profiling_instr.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_reduce_branching_through_overcompute.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_remove_assume.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_remove_no_op.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_remove_undef.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_remove_weight_layout_rewrite_block.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_renormalize_split_pattern.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_rewrite_unsafe_select.py (100%)
rename tests/python/{unittest => tir-transform}/test_tir_transform_simplify.py
(100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_split_host_device.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_storage_flatten.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_storage_rewrite.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_thread_sync.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_unify_thread_binding.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_unroll_loop.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_tir_transform_vectorize.py (100%)
rename tests/python/{unittest =>
tir-transform}/test_transform_default_gpu_schedule.py (100%)
rename tests/python/{unittest => tir-usmp}/test_tir_usmp_algo.py (100%)
rename tests/python/{unittest => tir-usmp}/test_tir_usmp_algo_hill_climb.py
(100%)
rename tests/python/{unittest =>
tir-usmp}/test_tir_usmp_analysis_extract_bufferinfo.py (100%)
rename tests/python/{unittest =>
tir-usmp}/test_tir_usmp_transform_convert_pool_allocations_to_offsets.py (100%)
rename tests/python/{unittest =>
tir-usmp}/test_tir_usmp_transform_create_io_allocates.py (100%)
rename tests/python/{unittest => tir-usmp}/test_tir_usmp_utils.py (100%)
rename tests/python/topi/{python => }/common.py (100%)
rename tests/python/topi/{python => }/test_fifo_buffer.py (100%)
rename tests/python/topi/{python => }/test_topi_argwhere.py (100%)
rename tests/python/topi/{python => }/test_topi_basic.py (100%)
rename tests/python/topi/{python => }/test_topi_batch_matmul.py (100%)
rename tests/python/topi/{python => }/test_topi_batch_matmul_tensorcore.py
(100%)
rename tests/python/topi/{python => }/test_topi_batch_norm.py (100%)
rename tests/python/topi/{python => }/test_topi_batch_to_space_nd.py (100%)
rename tests/python/topi/{python => }/test_topi_bitserial_conv2d.py (100%)
rename tests/python/topi/{python => }/test_topi_bitserial_conv2d_rasp.py (100%)
rename tests/python/topi/{python => }/test_topi_bitserial_dense.py (100%)
rename tests/python/topi/{python => }/test_topi_bnn.py (100%)
rename tests/python/topi/{python => }/test_topi_broadcast.py (100%)
rename tests/python/topi/{python => }/test_topi_clip.py (100%)
rename tests/python/topi/{python => }/test_topi_conv1d.py (100%)
rename tests/python/topi/{python => }/test_topi_conv1d_transpose_ncw.py (100%)
rename tests/python/topi/{python => }/test_topi_conv2d_NCHWc.py (100%)
rename tests/python/topi/{python => }/test_topi_conv2d_hwcn.py (100%)
rename tests/python/topi/{python => }/test_topi_conv2d_hwnc_tensorcore.py
(100%)
rename tests/python/topi/{python => }/test_topi_conv2d_int8.py (100%)
rename tests/python/topi/{python => }/test_topi_conv2d_nchw.py (100%)
rename tests/python/topi/{python => }/test_topi_conv2d_nhwc.py (100%)
rename tests/python/topi/{python => }/test_topi_conv2d_nhwc_pack_int8.py (100%)
rename tests/python/topi/{python => }/test_topi_conv2d_nhwc_tensorcore.py
(100%)
rename tests/python/topi/{python => }/test_topi_conv2d_nhwc_winograd.py (100%)
rename tests/python/topi/{python => }/test_topi_conv2d_tensordot_opts.py (100%)
rename tests/python/topi/{python => }/test_topi_conv2d_transpose_nchw.py (100%)
rename tests/python/topi/{python => }/test_topi_conv2d_winograd.py (100%)
rename tests/python/topi/{python => }/test_topi_conv3d_ncdhw.py (100%)
rename tests/python/topi/{python => }/test_topi_conv3d_ndhwc.py (100%)
rename tests/python/topi/{python => }/test_topi_conv3d_ndhwc_tensorcore.py
(100%)
rename tests/python/topi/{python => }/test_topi_conv3d_transpose_ncdhw.py
(100%)
rename tests/python/topi/{python => }/test_topi_conv3d_winograd.py (100%)
rename tests/python/topi/{python => }/test_topi_correlation.py (100%)
rename tests/python/topi/{python => }/test_topi_deformable_conv2d.py (100%)
rename tests/python/topi/{python => }/test_topi_dense.py (100%)
rename tests/python/topi/{python => }/test_topi_dense_tensorcore.py (100%)
rename tests/python/topi/{python => }/test_topi_depth_to_space.py (100%)
rename tests/python/topi/{python => }/test_topi_depthwise_conv2d.py (100%)
rename tests/python/topi/{python => }/test_topi_depthwise_conv2d_back_input.py
(100%)
rename tests/python/topi/{python =>
}/test_topi_depthwise_conv2d_back_weight.py (100%)
rename tests/python/topi/{python => }/test_topi_dft.py (100%)
rename tests/python/topi/{python => }/test_topi_dilate.py (100%)
rename tests/python/topi/{python => }/test_topi_einsum.py (100%)
rename tests/python/topi/{python => }/test_topi_group_conv2d.py (100%)
rename tests/python/topi/{python => }/test_topi_group_conv2d_NCHWc_int8.py
(100%)
rename tests/python/topi/{python => }/test_topi_group_conv2d_transpose.py
(100%)
rename tests/python/topi/{python => }/test_topi_group_norm.py (100%)
rename tests/python/topi/{python => }/test_topi_image.py (100%)
rename tests/python/topi/{python => }/test_topi_instance_norm.py (100%)
rename tests/python/topi/{python => }/test_topi_layer_norm.py (100%)
rename tests/python/topi/{python => }/test_topi_loss.py (100%)
rename tests/python/topi/{python => }/test_topi_lrn.py (100%)
rename tests/python/topi/{python => }/test_topi_lstm.py (100%)
rename tests/python/topi/{python => }/test_topi_math.py (100%)
rename tests/python/topi/{python => }/test_topi_matmul.py (65%)
rename tests/python/topi/{python => }/test_topi_pooling.py (100%)
rename tests/python/topi/{python => }/test_topi_prng.py (100%)
rename tests/python/topi/{python => }/test_topi_qnn.py (100%)
rename tests/python/topi/{python => }/test_topi_reduce.py (100%)
rename tests/python/topi/{python => }/test_topi_relu.py (100%)
rename tests/python/topi/{python => }/test_topi_reorg.py (100%)
rename tests/python/topi/{python => }/test_topi_rms_norm.py (83%)
rename tests/python/topi/{python => }/test_topi_scan.py (100%)
rename tests/python/topi/{python => }/test_topi_scatter.py (100%)
rename tests/python/topi/{python => }/test_topi_searchsorted.py (100%)
rename tests/python/topi/{python => }/test_topi_softmax.py (100%)
rename tests/python/topi/{python => }/test_topi_sort.py (100%)
rename tests/python/topi/{python => }/test_topi_space_to_batch_nd.py (100%)
rename tests/python/topi/{python => }/test_topi_space_to_depth.py (100%)
rename tests/python/topi/{python => }/test_topi_sparse.py (100%)
rename tests/python/topi/{python => }/test_topi_tensor.py (100%)
rename tests/python/topi/{python => }/test_topi_transform.py (100%)
rename tests/python/topi/{python => }/test_topi_unique.py (100%)
rename tests/python/topi/{python => }/test_topi_upsampling.py (100%)
rename tests/python/topi/{python => }/test_topi_util.py (100%)
rename tests/python/topi/{python => }/test_topi_vision.py (100%)
rename tests/python/{unittest => tvmscript}/test_tvmscript_complete.py (100%)
rename tests/python/{unittest => tvmscript}/test_tvmscript_error_report.py
(100%)
rename tests/python/{unittest => tvmscript}/test_tvmscript_ir_builder_base.py
(100%)
rename tests/python/{unittest =>
tvmscript}/test_tvmscript_ir_builder_irmodule.py (100%)
rename tests/python/{unittest => tvmscript}/test_tvmscript_ir_builder_tir.py
(99%)
rename tests/python/{unittest => tvmscript}/test_tvmscript_meta_programming.py
(100%)
rename tests/python/{unittest => tvmscript}/test_tvmscript_ops.py (100%)
rename tests/python/{unittest => tvmscript}/test_tvmscript_parser_evaluator.py
(100%)
rename tests/python/{unittest => tvmscript}/test_tvmscript_parser_ir.py (100%)
rename tests/python/{unittest => tvmscript}/test_tvmscript_parser_source.py
(100%)
rename tests/python/{unittest => tvmscript}/test_tvmscript_parser_tir.py (100%)
rename tests/python/{unittest =>
tvmscript}/test_tvmscript_printer_annotation.py (100%)
rename tests/python/{unittest => tvmscript}/test_tvmscript_printer_doc.py
(100%)
rename tests/python/{unittest =>
tvmscript}/test_tvmscript_printer_highlight.py (100%)
rename tests/python/{unittest => tvmscript}/test_tvmscript_printer_ir.py (100%)
rename tests/python/{unittest => tvmscript}/test_tvmscript_printer_metadata.py
(100%)
rename tests/python/{unittest =>
tvmscript}/test_tvmscript_printer_python_doc_printer.py (100%)
rename tests/python/{unittest =>
tvmscript}/test_tvmscript_printer_structural_equal.py (100%)
rename tests/python/{unittest => tvmscript}/test_tvmscript_printer_tir.py
(100%)
rename tests/python/{unittest =>
tvmscript}/test_tvmscript_printer_underlining.py (100%)
rename tests/python/{unittest => tvmscript}/test_tvmscript_regression.py (100%)
rename tests/python/{unittest => tvmscript}/test_tvmscript_roundtrip.py (99%)
rename tests/python/{unittest => tvmscript}/test_tvmscript_syntax_sugar.py
(100%)
rename tests/python/{unittest => tvmscript}/test_tvmscript_type.py (100%)
