(tvm) branch main updated: [TEST] Use subsystem-relative names for standalone TIRx tests (#20747)

[tqchen]

This is an automated email from the ASF dual-hosted git repository.

tqchen pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 3c64f417a2 [TEST] Use subsystem-relative names for standalone TIRx 
tests (#20747)
3c64f417a2 is described below

commit 3c64f417a21147de12c3573c3b3dc3f7391777d8
Author: Tianqi Chen <tqc...@users.noreply.github.com>
AuthorDate: Thu Oct 15 15:01:22 2026 -0400

    [TEST] Use subsystem-relative names for standalone TIRx tests (#20747)
---
 .agents/skills/tirx-ptx-dialect/SKILL.md           |  6 ++---
 .github/workflows/main.yml                         |  4 ++--
 docs/contribute/pull_request.rst                   |  2 +-
 tests/scripts/task_python_unittest_gpuonly.sh      |  4 ++--
 tirx/pyproject.toml                                |  2 +-
 tirx/python/tvm_tirx/tirx/_compile.py              |  2 +-
 .../__init__.py}                                   | 26 ----------------------
 .../cuda/__init__.py}                              | 26 ----------------------
 .../cuda/codegen/__init__.py}                      | 26 ----------------------
 .../cuda/codegen/{test_cuda.py => test_basic.py}   |  0
 ...est_cuda_host_bundle.py => test_host_bundle.py} |  0
 ...cuda_module_export.py => test_module_export.py} |  0
 .../hexagon/__init__.py}                           | 26 ----------------------
 .../hexagon/codegen/__init__.py}                   | 26 ----------------------
 .../codegen/{test_hexagon.py => test_basic.py}     |  0
 .../metal/__init__.py}                             | 26 ----------------------
 .../metal/codegen/__init__.py}                     | 26 ----------------------
 .../metal/codegen/{test_metal.py => test_basic.py} |  0
 ...ompile_callback.py => test_compile_callback.py} |  0
 ...etal_module_export.py => test_module_export.py} |  0
 .../opencl/__init__.py}                            | 26 ----------------------
 .../opencl/codegen/__init__.py}                    | 26 ----------------------
 .../codegen/{test_opencl.py => test_basic.py}      |  0
 ...encl_module_export.py => test_module_export.py} |  0
 .../rocm/__init__.py}                              | 26 ----------------------
 .../rocm/codegen/__init__.py}                      | 26 ----------------------
 .../rocm/codegen/{test_rocm.py => test_basic.py}   |  0
 ...rocm_module_export.py => test_module_export.py} |  0
 .../vulkan/__init__.py}                            | 26 ----------------------
 .../vulkan/codegen/__init__.py}                    | 26 ----------------------
 .../codegen/{test_vulkan.py => test_basic.py}      |  0
 ...lkan_module_export.py => test_module_export.py} |  0
 .../webgpu/__init__.py}                            | 26 ----------------------
 .../webgpu/codegen/__init__.py}                    | 26 ----------------------
 .../codegen/{test_webgpu.py => test_basic.py}      |  0
 .../__init__.py}                                   | 26 ----------------------
 .../c/__init__.py}                                 | 26 ----------------------
 .../python/codegen/c/{test_c.py => test_basic.py}  |  0
 .../gpu_common/__init__.py}                        | 26 ----------------------
 .../{test_gpu_intrinsics.py => test_intrinsics.py} |  0
 .../llvm/__init__.py}                              | 26 ----------------------
 .../codegen/llvm/{test_llvm.py => test_basic.py}   |  0
 .../__init__.py}                                   | 26 ----------------------
 .../python/ir/{test_ir_attrs.py => test_attrs.py}  |  0
 .../python/ir/{test_ir_json.py => test_json.py}    |  0
 .../ir/{test_ir_module.py => test_module.py}       |  0
 ...{test_nonnullable_ir.py => test_nonnullable.py} |  0
 ..._deep_equal.py => test_prim_expr_deep_equal.py} |  0
 .../python/ir/{test_ir_type.py => test_type.py}    |  0
 .../__init__.py}                                   | 26 ----------------------
 ...st_runtime_device_api.py => test_device_api.py} |  0
 ...untime_memory_scope.py => test_memory_scope.py} |  0
 .../{test_runtime_module.py => test_module.py}     |  0
 .../{test_runtime_tensor.py => test_tensor.py}     |  0
 ...t_script_printer_doc.py => test_printer_doc.py} |  0
 .../__init__.py}                                   | 26 ----------------------
 .../__init__.py}                                   | 26 ----------------------
 ..._analyzer_object.py => test_analyzer_object.py} |  0
 ..._const_int_bound.py => test_const_int_bound.py} |  0
 ...st_sym_deduce_bound.py => test_deduce_bound.py} |  0
 ...ect_clip_bound.py => test_detect_clip_bound.py} |  0
 ..._equation.py => test_detect_linear_equation.py} |  0
 .../sym/{test_sym_intset.py => test_intset.py}     |  0
 ..._iter_affine_map.py => test_iter_affine_map.py} |  0
 ...ewrite_simplify.py => test_rewrite_simplify.py} |  0
 .../sym/{test_sym_simplify.py => test_simplify.py} |  0
 .../python/sym/{test_sym_z3.py => test_z3.py}      |  0
 .../__init__.py}                                   | 26 ----------------------
 ..._parser_mprofile.py => test_parser_mprofile.py} |  0
 .../__init__.py}                                   | 26 ----------------------
 .../__init__.py}                                   | 26 ----------------------
 ..._well_defined.py => test_check_well_defined.py} |  0
 ...is_undefined_vars.py => test_undefined_vars.py} |  0
 .../__init__.py}                                   | 26 ----------------------
 .../{test_tirx_host_func.py => test_host_func.py}  |  0
 ...{test_tirx_imm_values.py => test_imm_values.py} |  0
 .../{test_tirx_index_map.py => test_index_map.py}  |  0
 .../base/{test_tirx_intrin.py => test_intrin.py}   |  0
 .../base/{test_tirx_nodes.py => test_nodes.py}     |  0
 .../{test_tirx_op_types.py => test_op_types.py}    |  0
 ...t_tirx_ptx_cp_async.py => test_ptx_cp_async.py} |  0
 ...riddepcontrol.py => test_ptx_griddepcontrol.py} |  0
 ...{test_tirx_ptx_mma_sp.py => test_ptx_mma_sp.py} |  0
 ...lar_f32_math.py => test_ptx_scalar_f32_math.py} |  0
 ...lable_datatype.py => test_scalable_datatype.py} |  0
 ...{test_tirx_specialize.py => test_specialize.py} |  0
 ...est_tirx_tensor_type.py => test_tensor_type.py} |  0
 ...tirx_texture_scope.py => test_texture_scope.py} |  0
 .../__init__.py}                                   | 26 ----------------------
 .../{test_codegen_ampere.py => test_ampere.py}     |  0
 ...test_codegen_blackwell.py => test_blackwell.py} |  0
 .../codegen/{test_codegen_cuda.py => test_cuda.py} |  0
 .../{test_codegen_dsmem.py => test_dsmem.py}       |  0
 .../{test_codegen_hopper.py => test_hopper.py}     |  0
 .../codegen/{test_codegen_nki.py => test_nki.py}   |  0
 .../__init__.py}                                   | 26 ----------------------
 .../oracle/__init__.py}                            | 26 ----------------------
 ...iket_orchestration.py => test_orchestration.py} |  0
 .../{test_iket_profiler.py => test_profiler.py}    |  0
 .../__init__.py}                                   | 26 ----------------------
 ...x_script_basic_usage.py => test_basic_usage.py} |  0
 ...ript_dynamic_shape.py => test_dynamic_shape.py} |  0
 ...pt_error_handling.py => test_error_handling.py} |  0
 .../{test_tirx_script_gpu.py => test_gpu.py}       |  0
 ...pecialization.py => test_jit_specialization.py} |  0
 ...eta_programming.py => test_meta_programming.py} |  0
 ...nter_highlight.py => test_printer_highlight.py} |  0
 ...info.py => test_printer_invisible_path_info.py} |  0
 ...l_equal.py => test_printer_structural_equal.py} |  0
 ..._underlining.py => test_printer_underlining.py} |  0
 ...ource_locations.py => test_source_locations.py} |  0
 ...test_tirx_script_tensors.py => test_tensors.py} |  0
 ...test_tirx_script_tracing.py => test_tracing.py} |  0
 .../{test_tirx_axis_enum.py => test_axis_enum.py}  |  0
 .../__init__.py}                                   | 26 ----------------------
 .../cuda/__init__.py}                              | 26 ----------------------
 .../trn/__init__.py}                               | 26 ----------------------
 .../trn/{test_binary_trn.py => test_binary.py}     |  0
 .../{test_compose_op_trn.py => test_compose_op.py} |  0
 .../tile/trn/{test_copy_trn.py => test_copy.py}    |  0
 .../tile/trn/{test_gemm_trn.py => test_gemm.py}    |  0
 ..._private_alloc_trn.py => test_private_alloc.py} |  0
 .../{test_reduction_trn.py => test_reduction.py}   |  0
 .../trn/{test_select_trn.py => test_select.py}     |  0
 .../tile/trn/{test_unary_trn.py => test_unary.py}  |  0
 ...nsform_decorate_device_scope.py => __init__.py} | 26 ----------------------
 ...form_bf16_legalize.py => test_bf16_legalize.py} |  0
 ...subexpr_elim.py => test_common_subexpr_elim.py} |  0
 ...ransform_convert_ssa.py => test_convert_ssa.py} |  0
 ...vice_scope.py => test_decorate_device_scope.py} |  0
 ...rm_flatten_tensor.py => test_flatten_tensor.py} |  0
 ...to_i32.py => test_force_narrow_index_to_i32.py} |  0
 ...nsform_fp8_legalize.py => test_fp8_legalize.py} |  0
 ...form_function_pass.py => test_function_pass.py} |  0
 ...t_tirx_transform_helpers.py => test_helpers.py} |  0
 ...tirx_transform_hoist_if.py => test_hoist_if.py} |  0
 ...nsform_lower_intrin.py => test_lower_intrin.py} |  0
 ..._transform_lower_tirx.py => test_lower_tirx.py} |  0
 ...er_tvm_builtin.py => test_lower_tvm_builtin.py} |  0
 ..._make_packed_api.py => test_make_packed_api.py} |  0
 ..._naive_allocator.py => test_naive_allocator.py} |  0
 ..._narrow_datatype.py => test_narrow_datatype.py} |  0
 ...write.py => test_pointer_value_type_rewrite.py} |  0
 ...nsform_remove_no_op.py => test_remove_no_op.py} |  0
 ...attern.py => test_renormalize_split_pattern.py} |  0
 ...tirx_transform_simplify.py => test_simplify.py} |  0
 ...it_host_device.py => test_split_host_device.py} |  0
 ..._storage_rewrite.py => test_storage_rewrite.py} |  0
 ...irx_transform_unroll_loop.py => test_unroll.py} |  0
 ...rx_transform_vectorize.py => test_vectorize.py} |  0
 150 files changed, 10 insertions(+), 920 deletions(-)

diff --git a/.agents/skills/tirx-ptx-dialect/SKILL.md 
b/.agents/skills/tirx-ptx-dialect/SKILL.md
index 5ffe3e6e00..e9213691b7 100644
--- a/.agents/skills/tirx-ptx-dialect/SKILL.md
+++ b/.agents/skills/tirx-ptx-dialect/SKILL.md
@@ -178,7 +178,7 @@ Run from the TVM repo root with the workspace `PYTHONPATH` 
(see `tir-test`).
      codegen test, since certification does not enumerate those helpers;
    - only when the instruction belongs to the MegaMoE extracted-intrinsics
      surface → its kernel and mnemonic list in
-     `test_codegen_cuda.py::test_megamoe_extracted_intrinsics_codegen`
+     
`tirx/tests/python/tirx/codegen/test_cuda.py::test_megamoe_extracted_intrinsics_codegen`
      (that test covers one workload, not the dialect).
 3. **Regenerate the stub** — any change to a family's tokens or shapes makes
    `test_ptx_stub_up_to_date` fail until you do:
@@ -277,8 +277,8 @@ Do this as one commit series, in this order:
    made the first time this table changed versions.
 3. **Retarget the citations.** Update the module docstring's version
    statement and URL, then run the renumbering over `table.py`,
-   `engine.py`, `render.py`, `tests/python/tirx/codegen/test_ptx_*.py` and
-   `test_codegen_cuda.py`, skipping any block that already carries the new
+   `engine.py`, `render.py`, `tirx/tests/python/tirx/codegen/test_ptx_*.py` and
+   `test_cuda.py`, skipping any block that already carries the new
    numbering (`_PTX_94_ENTRIES` was such a block for 9.3→9.4). Hand-fix the
    forms a regex cannot: ranges that span an insertion (`9.7.x.a-b`), brace
    groups (`9.7.4.{1,2,3}`), sibling shorthand (`.12`, `/ .17`), global
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 8603fdc95f..d3445bcc56 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -95,13 +95,13 @@ jobs:
         run: |
           set -euxo pipefail
           python -m pytest -v -s 
'tirx/tests/python/codegen/gpu_common/test_allreduce.py::test_allreduce_sum_compile'
-          python -m pytest -v -s 
'tirx/tests/python/backend/metal/codegen/test_metal.py::test_func_with_trailing_pod_params'
+          python -m pytest -v -s 
'tirx/tests/python/backend/metal/codegen/test_basic.py::test_func_with_trailing_pod_params'
       - name: Minimal Metal Compile-and-Run
         shell: bash -l {0}
         run: |
           set -euxo pipefail
           python -m pytest -v -s tirx/tests/python/backend/metal
-          python -m pytest -v -s 
'tirx/tests/python/codegen/gpu_common/test_gpu_intrinsics.py'
+          python -m pytest -v -s 
'tirx/tests/python/codegen/gpu_common/test_intrinsics.py'
           python -m pytest -v -s 
'tirx/tests/python/codegen/gpu_common/test_allreduce.py::test_allreduce_sum[dims0-metal]'
 
   Windows:
diff --git a/docs/contribute/pull_request.rst b/docs/contribute/pull_request.rst
index d349ce7d99..a93f55b546 100644
--- a/docs/contribute/pull_request.rst
+++ b/docs/contribute/pull_request.rst
@@ -261,7 +261,7 @@ If you want to run a single test:
   export PYTHONPATH=python
   rm -rf python/tvm/*.pyc python/tvm/*/*.pyc python/tvm/*/*/*.pyc
 
-  python -m pytest -v 
tests/python/tirx/transform/test_tirx_transform_storage_rewrite.py
+  python -m pytest -v tirx/tests/python/tirx/transform/test_storage_rewrite.py
 
   # Additionally if you want to run a single test, for example test_add_one_2d 
inside a file.
   python -m pytest -v -k "test_add_one_2d" 
tests/python/relax/test_frontend_tflite.py
diff --git a/tests/scripts/task_python_unittest_gpuonly.sh 
b/tests/scripts/task_python_unittest_gpuonly.sh
index c025bd6675..b891fdb3cd 100755
--- a/tests/scripts/task_python_unittest_gpuonly.sh
+++ b/tests/scripts/task_python_unittest_gpuonly.sh
@@ -35,5 +35,5 @@ export PYTHONPATH="$(pwd)/tirx/python:$(pwd)/python"
 export PYTEST_ADDOPTS="${CI_PYTEST_ADD_OPTIONS:-} ${PYTEST_ADDOPTS:-}"
 
 python3 -m pytest -vvs -n auto -m "${TVM_TEST_MARKER}" \
-    tirx/tests/python/backend/vulkan/codegen/test_vulkan.py \
-    tirx/tests/python/backend/vulkan/test_vulkan_module_export.py
+    tirx/tests/python/backend/vulkan/codegen/test_basic.py \
+    tirx/tests/python/backend/vulkan/test_module_export.py
diff --git a/tirx/pyproject.toml b/tirx/pyproject.toml
index db8d5c071c..1a5ae34d01 100644
--- a/tirx/pyproject.toml
+++ b/tirx/pyproject.toml
@@ -145,7 +145,7 @@ markers = [
 skip = "*-win32 *-manylinux_i686 *-musllinux*"
 build-verbosity = 1
 test-requires = ["pytest", "numpy"]
-test-command = "python -I -c \"import importlib.util; assert 
importlib.util.find_spec('tvm') is None\" && python -I -m pytest -vvs 
{package}/tests/python/all-platform-minimal-test 
{package}/tests/python/runtime/test_runtime_tensor.py 
{package}/tests/python/ir/test_ir_type.py 
{package}/tests/python/script/test_ir_builder.py 
{package}/tests/python/codegen/llvm/test_llvm.py::test_llvm_vadd_pipeline 
{package}/tests/python/codegen/gpu_common/test_allreduce.py::test_allreduce_sum_compile"
+test-command = "python -I -c \"import importlib.util; assert 
importlib.util.find_spec('tvm') is None\" && python -I -m pytest -vvs 
{package}/tests/python/all-platform-minimal-test 
{package}/tests/python/runtime/test_tensor.py 
{package}/tests/python/ir/test_type.py 
{package}/tests/python/script/test_ir_builder.py 
{package}/tests/python/codegen/llvm/test_basic.py::test_llvm_vadd_pipeline 
{package}/tests/python/codegen/gpu_common/test_allreduce.py::test_allreduce_sum_compile"
 
 [tool.cibuildwheel.linux]
 repair-wheel-command = "auditwheel repair --exclude libtvm_ffi.so --exclude 
libtvm_runtime_cuda.so --exclude 'libcuda.so.*' --exclude 'libcudart.so.*' 
--exclude 'libnvrtc.so.*' --exclude 'libnvrtc-builtins.so.*' -w {dest_dir} 
{wheel}"
diff --git a/tirx/python/tvm_tirx/tirx/_compile.py 
b/tirx/python/tvm_tirx/tirx/_compile.py
index 066d566f85..6d2a6f18b1 100644
--- a/tirx/python/tvm_tirx/tirx/_compile.py
+++ b/tirx/python/tvm_tirx/tirx/_compile.py
@@ -146,7 +146,7 @@ def _codegen_host_and_device(
     """Compile each device group and attach it to the host runtime module."""
 
     # Get the first module to get the attributes
-    # necessary for 
tests/python/backend/cuda/test_cuda_module_export.py::test_cuda_multi_lib
+    # necessary for 
tirx/tests/python/backend/cuda/test_module_export.py::test_cuda_multi_lib
     mhost_all = ir.IRModule({}, attrs=host_mod.attrs)
 
     mhost_all.update(host_mod)
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/backend/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/backend/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/backend/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/backend/cuda/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/backend/cuda/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/backend/cuda/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/backend/cuda/codegen/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/backend/cuda/codegen/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/backend/cuda/codegen/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git a/tirx/tests/python/backend/cuda/codegen/test_cuda.py 
b/tirx/tests/python/backend/cuda/codegen/test_basic.py
similarity index 100%
rename from tirx/tests/python/backend/cuda/codegen/test_cuda.py
rename to tirx/tests/python/backend/cuda/codegen/test_basic.py
diff --git a/tirx/tests/python/backend/cuda/test_cuda_host_bundle.py 
b/tirx/tests/python/backend/cuda/test_host_bundle.py
similarity index 100%
rename from tirx/tests/python/backend/cuda/test_cuda_host_bundle.py
rename to tirx/tests/python/backend/cuda/test_host_bundle.py
diff --git a/tirx/tests/python/backend/cuda/test_cuda_module_export.py 
b/tirx/tests/python/backend/cuda/test_module_export.py
similarity index 100%
rename from tirx/tests/python/backend/cuda/test_cuda_module_export.py
rename to tirx/tests/python/backend/cuda/test_module_export.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/backend/hexagon/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/backend/hexagon/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/backend/hexagon/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/backend/hexagon/codegen/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/backend/hexagon/codegen/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/backend/hexagon/codegen/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git a/tirx/tests/python/backend/hexagon/codegen/test_hexagon.py 
b/tirx/tests/python/backend/hexagon/codegen/test_basic.py
similarity index 100%
rename from tirx/tests/python/backend/hexagon/codegen/test_hexagon.py
rename to tirx/tests/python/backend/hexagon/codegen/test_basic.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/backend/metal/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/backend/metal/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/backend/metal/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/backend/metal/codegen/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/backend/metal/codegen/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/backend/metal/codegen/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git a/tirx/tests/python/backend/metal/codegen/test_metal.py 
b/tirx/tests/python/backend/metal/codegen/test_basic.py
similarity index 100%
rename from tirx/tests/python/backend/metal/codegen/test_metal.py
rename to tirx/tests/python/backend/metal/codegen/test_basic.py
diff --git a/tirx/tests/python/backend/metal/test_metal_compile_callback.py 
b/tirx/tests/python/backend/metal/test_compile_callback.py
similarity index 100%
rename from tirx/tests/python/backend/metal/test_metal_compile_callback.py
rename to tirx/tests/python/backend/metal/test_compile_callback.py
diff --git a/tirx/tests/python/backend/metal/test_metal_module_export.py 
b/tirx/tests/python/backend/metal/test_module_export.py
similarity index 100%
rename from tirx/tests/python/backend/metal/test_metal_module_export.py
rename to tirx/tests/python/backend/metal/test_module_export.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/backend/opencl/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/backend/opencl/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/backend/opencl/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/backend/opencl/codegen/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/backend/opencl/codegen/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/backend/opencl/codegen/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git a/tirx/tests/python/backend/opencl/codegen/test_opencl.py 
b/tirx/tests/python/backend/opencl/codegen/test_basic.py
similarity index 100%
rename from tirx/tests/python/backend/opencl/codegen/test_opencl.py
rename to tirx/tests/python/backend/opencl/codegen/test_basic.py
diff --git a/tirx/tests/python/backend/opencl/test_opencl_module_export.py 
b/tirx/tests/python/backend/opencl/test_module_export.py
similarity index 100%
rename from tirx/tests/python/backend/opencl/test_opencl_module_export.py
rename to tirx/tests/python/backend/opencl/test_module_export.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/backend/rocm/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/backend/rocm/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/backend/rocm/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/backend/rocm/codegen/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/backend/rocm/codegen/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/backend/rocm/codegen/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git a/tirx/tests/python/backend/rocm/codegen/test_rocm.py 
b/tirx/tests/python/backend/rocm/codegen/test_basic.py
similarity index 100%
rename from tirx/tests/python/backend/rocm/codegen/test_rocm.py
rename to tirx/tests/python/backend/rocm/codegen/test_basic.py
diff --git a/tirx/tests/python/backend/rocm/test_rocm_module_export.py 
b/tirx/tests/python/backend/rocm/test_module_export.py
similarity index 100%
rename from tirx/tests/python/backend/rocm/test_rocm_module_export.py
rename to tirx/tests/python/backend/rocm/test_module_export.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/backend/vulkan/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/backend/vulkan/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/backend/vulkan/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/backend/vulkan/codegen/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/backend/vulkan/codegen/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/backend/vulkan/codegen/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git a/tirx/tests/python/backend/vulkan/codegen/test_vulkan.py 
b/tirx/tests/python/backend/vulkan/codegen/test_basic.py
similarity index 100%
rename from tirx/tests/python/backend/vulkan/codegen/test_vulkan.py
rename to tirx/tests/python/backend/vulkan/codegen/test_basic.py
diff --git a/tirx/tests/python/backend/vulkan/test_vulkan_module_export.py 
b/tirx/tests/python/backend/vulkan/test_module_export.py
similarity index 100%
rename from tirx/tests/python/backend/vulkan/test_vulkan_module_export.py
rename to tirx/tests/python/backend/vulkan/test_module_export.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/backend/webgpu/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/backend/webgpu/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/backend/webgpu/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/backend/webgpu/codegen/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/backend/webgpu/codegen/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/backend/webgpu/codegen/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git a/tirx/tests/python/backend/webgpu/codegen/test_webgpu.py 
b/tirx/tests/python/backend/webgpu/codegen/test_basic.py
similarity index 100%
rename from tirx/tests/python/backend/webgpu/codegen/test_webgpu.py
rename to tirx/tests/python/backend/webgpu/codegen/test_basic.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/codegen/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/codegen/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/codegen/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/codegen/c/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/codegen/c/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/codegen/c/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git a/tirx/tests/python/codegen/c/test_c.py 
b/tirx/tests/python/codegen/c/test_basic.py
similarity index 100%
rename from tirx/tests/python/codegen/c/test_c.py
rename to tirx/tests/python/codegen/c/test_basic.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/codegen/gpu_common/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/codegen/gpu_common/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/codegen/gpu_common/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git a/tirx/tests/python/codegen/gpu_common/test_gpu_intrinsics.py 
b/tirx/tests/python/codegen/gpu_common/test_intrinsics.py
similarity index 100%
rename from tirx/tests/python/codegen/gpu_common/test_gpu_intrinsics.py
rename to tirx/tests/python/codegen/gpu_common/test_intrinsics.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/codegen/llvm/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/codegen/llvm/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/codegen/llvm/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git a/tirx/tests/python/codegen/llvm/test_llvm.py 
b/tirx/tests/python/codegen/llvm/test_basic.py
similarity index 100%
rename from tirx/tests/python/codegen/llvm/test_llvm.py
rename to tirx/tests/python/codegen/llvm/test_basic.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/ir/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/ir/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/ir/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git a/tirx/tests/python/ir/test_ir_attrs.py 
b/tirx/tests/python/ir/test_attrs.py
similarity index 100%
rename from tirx/tests/python/ir/test_ir_attrs.py
rename to tirx/tests/python/ir/test_attrs.py
diff --git a/tirx/tests/python/ir/test_ir_json.py 
b/tirx/tests/python/ir/test_json.py
similarity index 100%
rename from tirx/tests/python/ir/test_ir_json.py
rename to tirx/tests/python/ir/test_json.py
diff --git a/tirx/tests/python/ir/test_ir_module.py 
b/tirx/tests/python/ir/test_module.py
similarity index 100%
rename from tirx/tests/python/ir/test_ir_module.py
rename to tirx/tests/python/ir/test_module.py
diff --git a/tirx/tests/python/ir/test_nonnullable_ir.py 
b/tirx/tests/python/ir/test_nonnullable.py
similarity index 100%
rename from tirx/tests/python/ir/test_nonnullable_ir.py
rename to tirx/tests/python/ir/test_nonnullable.py
diff --git a/tirx/tests/python/ir/test_ir_prim_expr_deep_equal.py 
b/tirx/tests/python/ir/test_prim_expr_deep_equal.py
similarity index 100%
rename from tirx/tests/python/ir/test_ir_prim_expr_deep_equal.py
rename to tirx/tests/python/ir/test_prim_expr_deep_equal.py
diff --git a/tirx/tests/python/ir/test_ir_type.py 
b/tirx/tests/python/ir/test_type.py
similarity index 100%
rename from tirx/tests/python/ir/test_ir_type.py
rename to tirx/tests/python/ir/test_type.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/runtime/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/runtime/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/runtime/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git a/tirx/tests/python/runtime/test_runtime_device_api.py 
b/tirx/tests/python/runtime/test_device_api.py
similarity index 100%
rename from tirx/tests/python/runtime/test_runtime_device_api.py
rename to tirx/tests/python/runtime/test_device_api.py
diff --git a/tirx/tests/python/runtime/test_runtime_memory_scope.py 
b/tirx/tests/python/runtime/test_memory_scope.py
similarity index 100%
rename from tirx/tests/python/runtime/test_runtime_memory_scope.py
rename to tirx/tests/python/runtime/test_memory_scope.py
diff --git a/tirx/tests/python/runtime/test_runtime_module.py 
b/tirx/tests/python/runtime/test_module.py
similarity index 100%
rename from tirx/tests/python/runtime/test_runtime_module.py
rename to tirx/tests/python/runtime/test_module.py
diff --git a/tirx/tests/python/runtime/test_runtime_tensor.py 
b/tirx/tests/python/runtime/test_tensor.py
similarity index 100%
rename from tirx/tests/python/runtime/test_runtime_tensor.py
rename to tirx/tests/python/runtime/test_tensor.py
diff --git a/tirx/tests/python/script/test_script_printer_doc.py 
b/tirx/tests/python/script/test_printer_doc.py
similarity index 100%
rename from tirx/tests/python/script/test_script_printer_doc.py
rename to tirx/tests/python/script/test_printer_doc.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/support/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/support/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/support/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/sym/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/sym/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/sym/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git a/tirx/tests/python/sym/test_sym_analyzer_object.py 
b/tirx/tests/python/sym/test_analyzer_object.py
similarity index 100%
rename from tirx/tests/python/sym/test_sym_analyzer_object.py
rename to tirx/tests/python/sym/test_analyzer_object.py
diff --git a/tirx/tests/python/sym/test_sym_const_int_bound.py 
b/tirx/tests/python/sym/test_const_int_bound.py
similarity index 100%
rename from tirx/tests/python/sym/test_sym_const_int_bound.py
rename to tirx/tests/python/sym/test_const_int_bound.py
diff --git a/tirx/tests/python/sym/test_sym_deduce_bound.py 
b/tirx/tests/python/sym/test_deduce_bound.py
similarity index 100%
rename from tirx/tests/python/sym/test_sym_deduce_bound.py
rename to tirx/tests/python/sym/test_deduce_bound.py
diff --git a/tirx/tests/python/sym/test_sym_detect_clip_bound.py 
b/tirx/tests/python/sym/test_detect_clip_bound.py
similarity index 100%
rename from tirx/tests/python/sym/test_sym_detect_clip_bound.py
rename to tirx/tests/python/sym/test_detect_clip_bound.py
diff --git a/tirx/tests/python/sym/test_sym_detect_linear_equation.py 
b/tirx/tests/python/sym/test_detect_linear_equation.py
similarity index 100%
rename from tirx/tests/python/sym/test_sym_detect_linear_equation.py
rename to tirx/tests/python/sym/test_detect_linear_equation.py
diff --git a/tirx/tests/python/sym/test_sym_intset.py 
b/tirx/tests/python/sym/test_intset.py
similarity index 100%
rename from tirx/tests/python/sym/test_sym_intset.py
rename to tirx/tests/python/sym/test_intset.py
diff --git a/tirx/tests/python/sym/test_sym_iter_affine_map.py 
b/tirx/tests/python/sym/test_iter_affine_map.py
similarity index 100%
rename from tirx/tests/python/sym/test_sym_iter_affine_map.py
rename to tirx/tests/python/sym/test_iter_affine_map.py
diff --git a/tirx/tests/python/sym/test_sym_rewrite_simplify.py 
b/tirx/tests/python/sym/test_rewrite_simplify.py
similarity index 100%
rename from tirx/tests/python/sym/test_sym_rewrite_simplify.py
rename to tirx/tests/python/sym/test_rewrite_simplify.py
diff --git a/tirx/tests/python/sym/test_sym_simplify.py 
b/tirx/tests/python/sym/test_simplify.py
similarity index 100%
rename from tirx/tests/python/sym/test_sym_simplify.py
rename to tirx/tests/python/sym/test_simplify.py
diff --git a/tirx/tests/python/sym/test_sym_z3.py 
b/tirx/tests/python/sym/test_z3.py
similarity index 100%
rename from tirx/tests/python/sym/test_sym_z3.py
rename to tirx/tests/python/sym/test_z3.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/target/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/target/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/target/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git a/tirx/tests/python/target/test_target_parser_mprofile.py 
b/tirx/tests/python/target/test_parser_mprofile.py
similarity index 100%
rename from tirx/tests/python/target/test_target_parser_mprofile.py
rename to tirx/tests/python/target/test_parser_mprofile.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/testing/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/testing/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/testing/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/tirx/analysis/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/tirx/analysis/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/tirx/analysis/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git 
a/tirx/tests/python/tirx/analysis/test_tirx_analysis_check_well_defined.py 
b/tirx/tests/python/tirx/analysis/test_check_well_defined.py
similarity index 100%
rename from 
tirx/tests/python/tirx/analysis/test_tirx_analysis_check_well_defined.py
rename to tirx/tests/python/tirx/analysis/test_check_well_defined.py
diff --git 
a/tirx/tests/python/tirx/analysis/test_tirx_analysis_undefined_vars.py 
b/tirx/tests/python/tirx/analysis/test_undefined_vars.py
similarity index 100%
rename from tirx/tests/python/tirx/analysis/test_tirx_analysis_undefined_vars.py
rename to tirx/tests/python/tirx/analysis/test_undefined_vars.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/tirx/base/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/tirx/base/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/tirx/base/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git a/tirx/tests/python/tirx/base/test_tirx_host_func.py 
b/tirx/tests/python/tirx/base/test_host_func.py
similarity index 100%
rename from tirx/tests/python/tirx/base/test_tirx_host_func.py
rename to tirx/tests/python/tirx/base/test_host_func.py
diff --git a/tirx/tests/python/tirx/base/test_tirx_imm_values.py 
b/tirx/tests/python/tirx/base/test_imm_values.py
similarity index 100%
rename from tirx/tests/python/tirx/base/test_tirx_imm_values.py
rename to tirx/tests/python/tirx/base/test_imm_values.py
diff --git a/tirx/tests/python/tirx/base/test_tirx_index_map.py 
b/tirx/tests/python/tirx/base/test_index_map.py
similarity index 100%
rename from tirx/tests/python/tirx/base/test_tirx_index_map.py
rename to tirx/tests/python/tirx/base/test_index_map.py
diff --git a/tirx/tests/python/tirx/base/test_tirx_intrin.py 
b/tirx/tests/python/tirx/base/test_intrin.py
similarity index 100%
rename from tirx/tests/python/tirx/base/test_tirx_intrin.py
rename to tirx/tests/python/tirx/base/test_intrin.py
diff --git a/tirx/tests/python/tirx/base/test_tirx_nodes.py 
b/tirx/tests/python/tirx/base/test_nodes.py
similarity index 100%
rename from tirx/tests/python/tirx/base/test_tirx_nodes.py
rename to tirx/tests/python/tirx/base/test_nodes.py
diff --git a/tirx/tests/python/tirx/base/test_tirx_op_types.py 
b/tirx/tests/python/tirx/base/test_op_types.py
similarity index 100%
rename from tirx/tests/python/tirx/base/test_tirx_op_types.py
rename to tirx/tests/python/tirx/base/test_op_types.py
diff --git a/tirx/tests/python/tirx/base/test_tirx_ptx_cp_async.py 
b/tirx/tests/python/tirx/base/test_ptx_cp_async.py
similarity index 100%
rename from tirx/tests/python/tirx/base/test_tirx_ptx_cp_async.py
rename to tirx/tests/python/tirx/base/test_ptx_cp_async.py
diff --git a/tirx/tests/python/tirx/base/test_tirx_ptx_griddepcontrol.py 
b/tirx/tests/python/tirx/base/test_ptx_griddepcontrol.py
similarity index 100%
rename from tirx/tests/python/tirx/base/test_tirx_ptx_griddepcontrol.py
rename to tirx/tests/python/tirx/base/test_ptx_griddepcontrol.py
diff --git a/tirx/tests/python/tirx/base/test_tirx_ptx_mma_sp.py 
b/tirx/tests/python/tirx/base/test_ptx_mma_sp.py
similarity index 100%
rename from tirx/tests/python/tirx/base/test_tirx_ptx_mma_sp.py
rename to tirx/tests/python/tirx/base/test_ptx_mma_sp.py
diff --git a/tirx/tests/python/tirx/base/test_tirx_ptx_scalar_f32_math.py 
b/tirx/tests/python/tirx/base/test_ptx_scalar_f32_math.py
similarity index 100%
rename from tirx/tests/python/tirx/base/test_tirx_ptx_scalar_f32_math.py
rename to tirx/tests/python/tirx/base/test_ptx_scalar_f32_math.py
diff --git a/tirx/tests/python/tirx/base/test_tirx_scalable_datatype.py 
b/tirx/tests/python/tirx/base/test_scalable_datatype.py
similarity index 100%
rename from tirx/tests/python/tirx/base/test_tirx_scalable_datatype.py
rename to tirx/tests/python/tirx/base/test_scalable_datatype.py
diff --git a/tirx/tests/python/tirx/base/test_tirx_specialize.py 
b/tirx/tests/python/tirx/base/test_specialize.py
similarity index 100%
rename from tirx/tests/python/tirx/base/test_tirx_specialize.py
rename to tirx/tests/python/tirx/base/test_specialize.py
diff --git a/tirx/tests/python/tirx/base/test_tirx_tensor_type.py 
b/tirx/tests/python/tirx/base/test_tensor_type.py
similarity index 100%
rename from tirx/tests/python/tirx/base/test_tirx_tensor_type.py
rename to tirx/tests/python/tirx/base/test_tensor_type.py
diff --git a/tirx/tests/python/tirx/base/test_tirx_texture_scope.py 
b/tirx/tests/python/tirx/base/test_texture_scope.py
similarity index 100%
rename from tirx/tests/python/tirx/base/test_tirx_texture_scope.py
rename to tirx/tests/python/tirx/base/test_texture_scope.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/tirx/codegen/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/tirx/codegen/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/tirx/codegen/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git a/tirx/tests/python/tirx/codegen/test_codegen_ampere.py 
b/tirx/tests/python/tirx/codegen/test_ampere.py
similarity index 100%
rename from tirx/tests/python/tirx/codegen/test_codegen_ampere.py
rename to tirx/tests/python/tirx/codegen/test_ampere.py
diff --git a/tirx/tests/python/tirx/codegen/test_codegen_blackwell.py 
b/tirx/tests/python/tirx/codegen/test_blackwell.py
similarity index 100%
rename from tirx/tests/python/tirx/codegen/test_codegen_blackwell.py
rename to tirx/tests/python/tirx/codegen/test_blackwell.py
diff --git a/tirx/tests/python/tirx/codegen/test_codegen_cuda.py 
b/tirx/tests/python/tirx/codegen/test_cuda.py
similarity index 100%
rename from tirx/tests/python/tirx/codegen/test_codegen_cuda.py
rename to tirx/tests/python/tirx/codegen/test_cuda.py
diff --git a/tirx/tests/python/tirx/codegen/test_codegen_dsmem.py 
b/tirx/tests/python/tirx/codegen/test_dsmem.py
similarity index 100%
rename from tirx/tests/python/tirx/codegen/test_codegen_dsmem.py
rename to tirx/tests/python/tirx/codegen/test_dsmem.py
diff --git a/tirx/tests/python/tirx/codegen/test_codegen_hopper.py 
b/tirx/tests/python/tirx/codegen/test_hopper.py
similarity index 100%
rename from tirx/tests/python/tirx/codegen/test_codegen_hopper.py
rename to tirx/tests/python/tirx/codegen/test_hopper.py
diff --git a/tirx/tests/python/tirx/codegen/test_codegen_nki.py 
b/tirx/tests/python/tirx/codegen/test_nki.py
similarity index 100%
rename from tirx/tests/python/tirx/codegen/test_codegen_nki.py
rename to tirx/tests/python/tirx/codegen/test_nki.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/tirx/iket/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/tirx/iket/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/tirx/iket/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/tirx/iket/oracle/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/tirx/iket/oracle/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/tirx/iket/oracle/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git a/tirx/tests/python/tirx/iket/test_iket_orchestration.py 
b/tirx/tests/python/tirx/iket/test_orchestration.py
similarity index 100%
rename from tirx/tests/python/tirx/iket/test_iket_orchestration.py
rename to tirx/tests/python/tirx/iket/test_orchestration.py
diff --git a/tirx/tests/python/tirx/iket/test_iket_profiler.py 
b/tirx/tests/python/tirx/iket/test_profiler.py
similarity index 100%
rename from tirx/tests/python/tirx/iket/test_iket_profiler.py
rename to tirx/tests/python/tirx/iket/test_profiler.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/tirx/script/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/tirx/script/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/tirx/script/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git a/tirx/tests/python/tirx/script/test_tirx_script_basic_usage.py 
b/tirx/tests/python/tirx/script/test_basic_usage.py
similarity index 100%
rename from tirx/tests/python/tirx/script/test_tirx_script_basic_usage.py
rename to tirx/tests/python/tirx/script/test_basic_usage.py
diff --git a/tirx/tests/python/tirx/script/test_tirx_script_dynamic_shape.py 
b/tirx/tests/python/tirx/script/test_dynamic_shape.py
similarity index 100%
rename from tirx/tests/python/tirx/script/test_tirx_script_dynamic_shape.py
rename to tirx/tests/python/tirx/script/test_dynamic_shape.py
diff --git a/tirx/tests/python/tirx/script/test_tirx_script_error_handling.py 
b/tirx/tests/python/tirx/script/test_error_handling.py
similarity index 100%
rename from tirx/tests/python/tirx/script/test_tirx_script_error_handling.py
rename to tirx/tests/python/tirx/script/test_error_handling.py
diff --git a/tirx/tests/python/tirx/script/test_tirx_script_gpu.py 
b/tirx/tests/python/tirx/script/test_gpu.py
similarity index 100%
rename from tirx/tests/python/tirx/script/test_tirx_script_gpu.py
rename to tirx/tests/python/tirx/script/test_gpu.py
diff --git 
a/tirx/tests/python/tirx/script/test_tirx_script_jit_specialization.py 
b/tirx/tests/python/tirx/script/test_jit_specialization.py
similarity index 100%
rename from tirx/tests/python/tirx/script/test_tirx_script_jit_specialization.py
rename to tirx/tests/python/tirx/script/test_jit_specialization.py
diff --git a/tirx/tests/python/tirx/script/test_tirx_script_meta_programming.py 
b/tirx/tests/python/tirx/script/test_meta_programming.py
similarity index 100%
rename from tirx/tests/python/tirx/script/test_tirx_script_meta_programming.py
rename to tirx/tests/python/tirx/script/test_meta_programming.py
diff --git 
a/tirx/tests/python/tirx/script/test_tirx_script_printer_highlight.py 
b/tirx/tests/python/tirx/script/test_printer_highlight.py
similarity index 100%
rename from tirx/tests/python/tirx/script/test_tirx_script_printer_highlight.py
rename to tirx/tests/python/tirx/script/test_printer_highlight.py
diff --git 
a/tirx/tests/python/tirx/script/test_tirx_script_printer_invisible_path_info.py 
b/tirx/tests/python/tirx/script/test_printer_invisible_path_info.py
similarity index 100%
rename from 
tirx/tests/python/tirx/script/test_tirx_script_printer_invisible_path_info.py
rename to tirx/tests/python/tirx/script/test_printer_invisible_path_info.py
diff --git 
a/tirx/tests/python/tirx/script/test_tirx_script_printer_structural_equal.py 
b/tirx/tests/python/tirx/script/test_printer_structural_equal.py
similarity index 100%
rename from 
tirx/tests/python/tirx/script/test_tirx_script_printer_structural_equal.py
rename to tirx/tests/python/tirx/script/test_printer_structural_equal.py
diff --git 
a/tirx/tests/python/tirx/script/test_tirx_script_printer_underlining.py 
b/tirx/tests/python/tirx/script/test_printer_underlining.py
similarity index 100%
rename from 
tirx/tests/python/tirx/script/test_tirx_script_printer_underlining.py
rename to tirx/tests/python/tirx/script/test_printer_underlining.py
diff --git a/tirx/tests/python/tirx/script/test_tirx_script_source_locations.py 
b/tirx/tests/python/tirx/script/test_source_locations.py
similarity index 100%
rename from tirx/tests/python/tirx/script/test_tirx_script_source_locations.py
rename to tirx/tests/python/tirx/script/test_source_locations.py
diff --git a/tirx/tests/python/tirx/script/test_tirx_script_tensors.py 
b/tirx/tests/python/tirx/script/test_tensors.py
similarity index 100%
rename from tirx/tests/python/tirx/script/test_tirx_script_tensors.py
rename to tirx/tests/python/tirx/script/test_tensors.py
diff --git a/tirx/tests/python/tirx/script/test_tirx_script_tracing.py 
b/tirx/tests/python/tirx/script/test_tracing.py
similarity index 100%
rename from tirx/tests/python/tirx/script/test_tirx_script_tracing.py
rename to tirx/tests/python/tirx/script/test_tracing.py
diff --git a/tirx/tests/python/tirx/test_tirx_axis_enum.py 
b/tirx/tests/python/tirx/test_axis_enum.py
similarity index 100%
rename from tirx/tests/python/tirx/test_tirx_axis_enum.py
rename to tirx/tests/python/tirx/test_axis_enum.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/tirx/tile/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/tirx/tile/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/tirx/tile/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/tirx/tile/cuda/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/tirx/tile/cuda/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/tirx/tile/cuda/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/tirx/tile/trn/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/tirx/tile/trn/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/tirx/tile/trn/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git a/tirx/tests/python/tirx/tile/trn/test_binary_trn.py 
b/tirx/tests/python/tirx/tile/trn/test_binary.py
similarity index 100%
rename from tirx/tests/python/tirx/tile/trn/test_binary_trn.py
rename to tirx/tests/python/tirx/tile/trn/test_binary.py
diff --git a/tirx/tests/python/tirx/tile/trn/test_compose_op_trn.py 
b/tirx/tests/python/tirx/tile/trn/test_compose_op.py
similarity index 100%
rename from tirx/tests/python/tirx/tile/trn/test_compose_op_trn.py
rename to tirx/tests/python/tirx/tile/trn/test_compose_op.py
diff --git a/tirx/tests/python/tirx/tile/trn/test_copy_trn.py 
b/tirx/tests/python/tirx/tile/trn/test_copy.py
similarity index 100%
rename from tirx/tests/python/tirx/tile/trn/test_copy_trn.py
rename to tirx/tests/python/tirx/tile/trn/test_copy.py
diff --git a/tirx/tests/python/tirx/tile/trn/test_gemm_trn.py 
b/tirx/tests/python/tirx/tile/trn/test_gemm.py
similarity index 100%
rename from tirx/tests/python/tirx/tile/trn/test_gemm_trn.py
rename to tirx/tests/python/tirx/tile/trn/test_gemm.py
diff --git a/tirx/tests/python/tirx/tile/trn/test_private_alloc_trn.py 
b/tirx/tests/python/tirx/tile/trn/test_private_alloc.py
similarity index 100%
rename from tirx/tests/python/tirx/tile/trn/test_private_alloc_trn.py
rename to tirx/tests/python/tirx/tile/trn/test_private_alloc.py
diff --git a/tirx/tests/python/tirx/tile/trn/test_reduction_trn.py 
b/tirx/tests/python/tirx/tile/trn/test_reduction.py
similarity index 100%
rename from tirx/tests/python/tirx/tile/trn/test_reduction_trn.py
rename to tirx/tests/python/tirx/tile/trn/test_reduction.py
diff --git a/tirx/tests/python/tirx/tile/trn/test_select_trn.py 
b/tirx/tests/python/tirx/tile/trn/test_select.py
similarity index 100%
rename from tirx/tests/python/tirx/tile/trn/test_select_trn.py
rename to tirx/tests/python/tirx/tile/trn/test_select.py
diff --git a/tirx/tests/python/tirx/tile/trn/test_unary_trn.py 
b/tirx/tests/python/tirx/tile/trn/test_unary.py
similarity index 100%
rename from tirx/tests/python/tirx/tile/trn/test_unary_trn.py
rename to tirx/tests/python/tirx/tile/trn/test_unary.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/tirx/transform/__init__.py
similarity index 55%
copy from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
copy to tirx/tests/python/tirx/transform/__init__.py
index a856fc0ccd..13a83393a9 100644
--- 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
+++ b/tirx/tests/python/tirx/transform/__init__.py
@@ -14,29 +14,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm_tirx
-import tvm_tirx.testing
-from tvm_tirx.script import ir as I
-from tvm_tirx.script import tirx as T
-
-
-def test_decorate_device():
-    @I.ir_module
-    class Before:
-        @T.function(private=True)
-        def main(x: T.int32):
-            T.evaluate(x)
-
-    @I.ir_module
-    class Expected:
-        @T.function(private=True)
-        def main(x: T.int32):
-            with T.region("tirx.device_scope", []):
-                T.evaluate(x)
-
-    After = tvm_tirx.tirx.transform.DecorateDeviceScope()(Before)
-    tvm_tirx.ir.assert_structural_equal(After, Expected)
-
-
-if __name__ == "__main__":
-    tvm_tirx.testing.main()
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_bf16_legalize.py 
b/tirx/tests/python/tirx/transform/test_bf16_legalize.py
similarity index 100%
rename from 
tirx/tests/python/tirx/transform/test_tirx_transform_bf16_legalize.py
rename to tirx/tests/python/tirx/transform/test_bf16_legalize.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_common_subexpr_elim.py 
b/tirx/tests/python/tirx/transform/test_common_subexpr_elim.py
similarity index 100%
rename from 
tirx/tests/python/tirx/transform/test_tirx_transform_common_subexpr_elim.py
rename to tirx/tests/python/tirx/transform/test_common_subexpr_elim.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_convert_ssa.py 
b/tirx/tests/python/tirx/transform/test_convert_ssa.py
similarity index 100%
rename from tirx/tests/python/tirx/transform/test_tirx_transform_convert_ssa.py
rename to tirx/tests/python/tirx/transform/test_convert_ssa.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py 
b/tirx/tests/python/tirx/transform/test_decorate_device_scope.py
similarity index 100%
rename from 
tirx/tests/python/tirx/transform/test_tirx_transform_decorate_device_scope.py
rename to tirx/tests/python/tirx/transform/test_decorate_device_scope.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_flatten_tensor.py 
b/tirx/tests/python/tirx/transform/test_flatten_tensor.py
similarity index 100%
rename from 
tirx/tests/python/tirx/transform/test_tirx_transform_flatten_tensor.py
rename to tirx/tests/python/tirx/transform/test_flatten_tensor.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_force_narrow_index_to_i32.py
 b/tirx/tests/python/tirx/transform/test_force_narrow_index_to_i32.py
similarity index 100%
rename from 
tirx/tests/python/tirx/transform/test_tirx_transform_force_narrow_index_to_i32.py
rename to tirx/tests/python/tirx/transform/test_force_narrow_index_to_i32.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_fp8_legalize.py 
b/tirx/tests/python/tirx/transform/test_fp8_legalize.py
similarity index 100%
rename from tirx/tests/python/tirx/transform/test_tirx_transform_fp8_legalize.py
rename to tirx/tests/python/tirx/transform/test_fp8_legalize.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_function_pass.py 
b/tirx/tests/python/tirx/transform/test_function_pass.py
similarity index 100%
rename from 
tirx/tests/python/tirx/transform/test_tirx_transform_function_pass.py
rename to tirx/tests/python/tirx/transform/test_function_pass.py
diff --git a/tirx/tests/python/tirx/transform/test_tirx_transform_helpers.py 
b/tirx/tests/python/tirx/transform/test_helpers.py
similarity index 100%
rename from tirx/tests/python/tirx/transform/test_tirx_transform_helpers.py
rename to tirx/tests/python/tirx/transform/test_helpers.py
diff --git a/tirx/tests/python/tirx/transform/test_tirx_transform_hoist_if.py 
b/tirx/tests/python/tirx/transform/test_hoist_if.py
similarity index 100%
rename from tirx/tests/python/tirx/transform/test_tirx_transform_hoist_if.py
rename to tirx/tests/python/tirx/transform/test_hoist_if.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_lower_intrin.py 
b/tirx/tests/python/tirx/transform/test_lower_intrin.py
similarity index 100%
rename from tirx/tests/python/tirx/transform/test_tirx_transform_lower_intrin.py
rename to tirx/tests/python/tirx/transform/test_lower_intrin.py
diff --git a/tirx/tests/python/tirx/transform/test_tirx_transform_lower_tirx.py 
b/tirx/tests/python/tirx/transform/test_lower_tirx.py
similarity index 100%
rename from tirx/tests/python/tirx/transform/test_tirx_transform_lower_tirx.py
rename to tirx/tests/python/tirx/transform/test_lower_tirx.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_lower_tvm_builtin.py 
b/tirx/tests/python/tirx/transform/test_lower_tvm_builtin.py
similarity index 100%
rename from 
tirx/tests/python/tirx/transform/test_tirx_transform_lower_tvm_builtin.py
rename to tirx/tests/python/tirx/transform/test_lower_tvm_builtin.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_make_packed_api.py 
b/tirx/tests/python/tirx/transform/test_make_packed_api.py
similarity index 100%
rename from 
tirx/tests/python/tirx/transform/test_tirx_transform_make_packed_api.py
rename to tirx/tests/python/tirx/transform/test_make_packed_api.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_naive_allocator.py 
b/tirx/tests/python/tirx/transform/test_naive_allocator.py
similarity index 100%
rename from 
tirx/tests/python/tirx/transform/test_tirx_transform_naive_allocator.py
rename to tirx/tests/python/tirx/transform/test_naive_allocator.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_narrow_datatype.py 
b/tirx/tests/python/tirx/transform/test_narrow_datatype.py
similarity index 100%
rename from 
tirx/tests/python/tirx/transform/test_tirx_transform_narrow_datatype.py
rename to tirx/tests/python/tirx/transform/test_narrow_datatype.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_pointer_value_type_rewrite.py
 b/tirx/tests/python/tirx/transform/test_pointer_value_type_rewrite.py
similarity index 100%
rename from 
tirx/tests/python/tirx/transform/test_tirx_transform_pointer_value_type_rewrite.py
rename to tirx/tests/python/tirx/transform/test_pointer_value_type_rewrite.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_remove_no_op.py 
b/tirx/tests/python/tirx/transform/test_remove_no_op.py
similarity index 100%
rename from tirx/tests/python/tirx/transform/test_tirx_transform_remove_no_op.py
rename to tirx/tests/python/tirx/transform/test_remove_no_op.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_renormalize_split_pattern.py
 b/tirx/tests/python/tirx/transform/test_renormalize_split_pattern.py
similarity index 100%
rename from 
tirx/tests/python/tirx/transform/test_tirx_transform_renormalize_split_pattern.py
rename to tirx/tests/python/tirx/transform/test_renormalize_split_pattern.py
diff --git a/tirx/tests/python/tirx/transform/test_tirx_transform_simplify.py 
b/tirx/tests/python/tirx/transform/test_simplify.py
similarity index 100%
rename from tirx/tests/python/tirx/transform/test_tirx_transform_simplify.py
rename to tirx/tests/python/tirx/transform/test_simplify.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_split_host_device.py 
b/tirx/tests/python/tirx/transform/test_split_host_device.py
similarity index 100%
rename from 
tirx/tests/python/tirx/transform/test_tirx_transform_split_host_device.py
rename to tirx/tests/python/tirx/transform/test_split_host_device.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_storage_rewrite.py 
b/tirx/tests/python/tirx/transform/test_storage_rewrite.py
similarity index 100%
rename from 
tirx/tests/python/tirx/transform/test_tirx_transform_storage_rewrite.py
rename to tirx/tests/python/tirx/transform/test_storage_rewrite.py
diff --git 
a/tirx/tests/python/tirx/transform/test_tirx_transform_unroll_loop.py 
b/tirx/tests/python/tirx/transform/test_unroll.py
similarity index 100%
rename from tirx/tests/python/tirx/transform/test_tirx_transform_unroll_loop.py
rename to tirx/tests/python/tirx/transform/test_unroll.py
diff --git a/tirx/tests/python/tirx/transform/test_tirx_transform_vectorize.py 
b/tirx/tests/python/tirx/transform/test_vectorize.py
similarity index 100%
rename from tirx/tests/python/tirx/transform/test_tirx_transform_vectorize.py
rename to tirx/tests/python/tirx/transform/test_vectorize.py