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 c414e4ca36 [chore][TIR] reorganize src/tir/transforms to
src/tir/transform (#18738)
c414e4ca36 is described below
commit c414e4ca36a9e3b52543183f69a369abbe944136
Author: Tianqi Chen <[email protected]>
AuthorDate: Mon Feb 9 10:28:54 2026 -0500
[chore][TIR] reorganize src/tir/transforms to src/tir/transform (#18738)
This PR reorganizes src/tir/transforms to src/tir/transform to be
consistent with the python namespace.
---
docs/arch/pass_infra.rst | 4 ++--
src/arith/detect_common_subexpr.cc | 2 +-
src/arith/int_constraints.cc | 2 +-
src/arith/scalable_expression.cc | 2 +-
src/s_tir/schedule/primitive/block_annotate.cc | 2 +-
src/s_tir/schedule/primitive/blockize_tensorize.cc | 2 +-
src/s_tir/schedule/primitive/cache_index.cc | 4 ++--
src/s_tir/schedule/primitive/cache_read_write.cc | 2 +-
src/s_tir/schedule/primitive/decompose_padding.cc | 2 +-
src/s_tir/schedule/primitive/hide_buffer_access.cc | 2 +-
src/s_tir/schedule/transform.cc | 2 +-
src/s_tir/transform/canonicalize_loop.cc | 2 +-
src/s_tir/transform/compact_buffer_region.cc | 2 +-
src/s_tir/transform/convert_blocks_to_opaque.cc | 2 +-
src/s_tir/transform/inject_double_buffer.cc | 2 +-
src/s_tir/transform/inject_permuted_layout.cc | 2 +-
src/s_tir/transform/inject_software_pipeline.cc | 2 +-
src/s_tir/transform/inject_virtual_thread.cc | 2 +-
src/s_tir/transform/lift_thread_binding.cc | 2 +-
src/s_tir/transform/loop_partition.cc | 2 +-
src/s_tir/transform/lower_cross_thread_reduction.cc | 2 +-
src/s_tir/transform/lower_init_block.cc | 2 +-
src/s_tir/transform/lower_match_buffer.cc | 2 +-
src/s_tir/transform/lower_opaque_block.cc | 2 +-
src/s_tir/transform/memhammer_lower_auto_copy.cc | 2 +-
src/s_tir/transform/plan_update_buffer_allocation_location.cc | 2 +-
src/s_tir/transform/transform_mma_buffer_layout.cc | 2 +-
src/s_tir/transform/unify_thread_binding.cc | 2 +-
src/script/printer/tir/stmt.cc | 2 +-
src/target/llvm/codegen_llvm.h | 2 +-
src/target/source/codegen_c.h | 2 +-
src/target/source/codegen_cuda.cc | 2 +-
src/target/spirv/codegen_spirv.cc | 2 +-
src/te/operation/create_primfunc.cc | 2 +-
src/tir/analysis/block_access_region_detector.cc | 2 +-
src/tir/analysis/verify_gpu_code.cc | 2 +-
src/tir/backend/adreno/inject_texture_alloc.cc | 2 +-
src/tir/ir/specialize.cc | 2 +-
src/tir/{transforms => transform}/annotate_device_regions.cc | 0
src/tir/{transforms => transform}/arg_binder.cc | 0
src/tir/{transforms => transform}/arg_binder.h | 6 +++---
src/tir/{transforms => transform}/bind_params.cc | 0
src/tir/{transforms => transform}/bind_target.cc | 0
src/tir/{transforms => transform}/bound_checker.cc | 0
src/tir/{transforms => transform}/combine_context_call.cc | 0
src/tir/{transforms => transform}/common_subexpr_elim.cc | 0
src/tir/{transforms => transform}/common_subexpr_elim.h | 6 +++---
src/tir/{transforms => transform}/common_subexpr_elim_tools.cc | 0
src/tir/{transforms => transform}/common_subexpr_elim_tools.h | 6 +++---
src/tir/{transforms => transform}/convert_for_loops_serial.cc | 2 +-
src/tir/{transforms => transform}/decorate_device_scope.cc | 0
src/tir/{transforms => transform}/default_gpu_schedule.cc | 0
src/tir/{transforms => transform}/dtype_conversion.cc | 0
src/tir/{transforms => transform}/dtype_conversion.h | 6 +++---
src/tir/{transforms => transform}/extract_constants.cc | 0
src/tir/{transforms => transform}/flatten_buffer.cc | 0
src/tir/{transforms => transform}/force_narrow_index_to_i32.cc | 0
src/tir/{transforms => transform}/hoist_expression.cc | 0
src/tir/{transforms => transform}/inject_ptx_async_copy.cc | 0
src/tir/{transforms => transform}/inject_ptx_ldg32.cc | 0
src/tir/{transforms => transform}/inject_rolling_buffer.cc | 0
src/tir/{transforms => transform}/inline_private_functions.cc | 0
src/tir/{transforms => transform}/ir_utils.cc | 0
src/tir/{transforms => transform}/ir_utils.h | 6 +++---
src/tir/{transforms => transform}/lower_async_dma.cc | 0
src/tir/{transforms => transform}/lower_custom_datatypes.cc | 0
src/tir/{transforms => transform}/lower_device_kernel_launch.cc | 0
.../{transforms => transform}/lower_device_storage_access_info.cc | 0
src/tir/{transforms => transform}/lower_intrin.cc | 0
src/tir/{transforms => transform}/lower_thread_allreduce.cc | 0
src/tir/{transforms => transform}/lower_tvm_builtin.cc | 2 +-
src/tir/{transforms => transform}/lower_vtcm_alloc.cc | 0
src/tir/{transforms => transform}/lower_warp_memory.cc | 0
src/tir/{transforms => transform}/make_packed_api.cc | 0
src/tir/{transforms => transform}/make_unpacked_api.cc | 0
.../{transforms => transform}/merge_shared_memory_allocations.cc | 0
src/tir/{transforms => transform}/narrow_datatype.cc | 0
src/tir/{transforms => transform}/primfunc_utils.cc | 0
src/tir/{transforms => transform}/profile_instrumentation.cc | 0
.../reduce_branching_through_overcompute.cc | 0
src/tir/{transforms => transform}/remap_thread_axis.cc | 0
src/tir/{transforms => transform}/remove_assume.cc | 0
src/tir/{transforms => transform}/remove_no_op.cc | 0
src/tir/{transforms => transform}/remove_no_op.h | 6 +++---
src/tir/{transforms => transform}/remove_store_undef.cc | 0
.../{transforms => transform}/remove_weight_layout_rewrite_block.cc | 0
src/tir/{transforms => transform}/renew_defs.cc | 0
src/tir/{transforms => transform}/renormalize_split_pattern.cc | 0
src/tir/{transforms => transform}/replace_global_vars.cc | 2 +-
src/tir/{transforms => transform}/replace_selected_expr.cc | 0
src/tir/{transforms => transform}/replace_selected_expr.h | 6 +++---
src/tir/{transforms => transform}/rewrite_unsafe_select.cc | 0
src/tir/{transforms => transform}/simplify.cc | 2 +-
src/tir/{transforms => transform}/simplify.h | 6 +++---
src/tir/{transforms => transform}/skip_assert.cc | 0
src/tir/{transforms => transform}/split_host_device.cc | 0
src/tir/{transforms => transform}/storage_access.cc | 0
src/tir/{transforms => transform}/storage_access.h | 6 +++---
src/tir/{transforms => transform}/storage_rewrite.cc | 0
src/tir/{transforms => transform}/tensorcore_infer_fragment.cc | 0
src/tir/{transforms => transform}/thread_storage_sync.cc | 0
src/tir/{transforms => transform}/unroll_loop.cc | 0
src/tir/{transforms => transform}/unsupported_dtype_legalize.cc | 0
src/tir/{transforms => transform}/update_pointer_storage_scope.cc | 0
src/tir/{transforms => transform}/update_pointer_storage_scope.h | 6 +++---
.../{transforms => transform}/using_assume_to_reduce_branches.cc | 0
src/tir/{transforms => transform}/vectorize_loop.cc | 0
107 files changed, 74 insertions(+), 74 deletions(-)
diff --git a/docs/arch/pass_infra.rst b/docs/arch/pass_infra.rst
index ef3672058c..07f267c54c 100644
--- a/docs/arch/pass_infra.rst
+++ b/docs/arch/pass_infra.rst
@@ -578,7 +578,7 @@ loop unrolling pass
TVM_REGISTER_PASS_CONFIG_OPTION("tir.UnrollLoop", UnrollLoopConfig);
-Please refer to `src/tir/transforms/unroll_loop.cc`_ for more details.
+Please refer to `src/tir/transform/unroll_loop.cc`_ for more details.
.. _pass_instrument_py_frontend:
@@ -661,7 +661,7 @@ new ``PassInstrument`` are called.
.. _python/tvm/ir/instrument.py:
https://github.com/apache/tvm/blob/main/python/tvm/ir/instrument.py
-.. _src/tir/transforms/unroll_loop.cc:
https://github.com/apache/tvm/blob/main/src/tir/transforms/unroll_loop.cc
+.. _src/tir/transform/unroll_loop.cc:
https://github.com/apache/tvm/blob/main/src/tir/transform/unroll_loop.cc
.. _use pass infra:
https://github.com/apache/tvm/blob/main/tutorials/dev/use_pass_infra.py
diff --git a/src/arith/detect_common_subexpr.cc
b/src/arith/detect_common_subexpr.cc
index 70768128e5..ce7f1ec4c5 100644
--- a/src/arith/detect_common_subexpr.cc
+++ b/src/arith/detect_common_subexpr.cc
@@ -26,7 +26,7 @@
#include <limits>
-#include "../tir/transforms/common_subexpr_elim_tools.h"
+#include "../tir/transform/common_subexpr_elim_tools.h"
namespace tvm {
namespace arith {
diff --git a/src/arith/int_constraints.cc b/src/arith/int_constraints.cc
index e116ba9e3b..81c845906c 100644
--- a/src/arith/int_constraints.cc
+++ b/src/arith/int_constraints.cc
@@ -34,7 +34,7 @@
#include <unordered_map>
#include <utility>
-#include "../tir/transforms/ir_utils.h"
+#include "../tir/transform/ir_utils.h"
namespace tvm {
namespace arith {
diff --git a/src/arith/scalable_expression.cc b/src/arith/scalable_expression.cc
index 5c968966e2..c4e26d0725 100644
--- a/src/arith/scalable_expression.cc
+++ b/src/arith/scalable_expression.cc
@@ -30,7 +30,7 @@
#include <vector>
#include "../tir/analysis/check_contains.h"
-#include "../tir/transforms/replace_selected_expr.h"
+#include "../tir/transform/replace_selected_expr.h"
#include "./pattern_match.h"
namespace tvm {
diff --git a/src/s_tir/schedule/primitive/block_annotate.cc
b/src/s_tir/schedule/primitive/block_annotate.cc
index 0d6b2a58d0..ba8b8ccb83 100644
--- a/src/s_tir/schedule/primitive/block_annotate.cc
+++ b/src/s_tir/schedule/primitive/block_annotate.cc
@@ -19,7 +19,7 @@
#include <tvm/ffi/container/tuple.h>
#include <tvm/tir/expr.h>
-#include "../../../tir/transforms/ir_utils.h"
+#include "../../../tir/transform/ir_utils.h"
#include "../utils.h"
namespace tvm {
diff --git a/src/s_tir/schedule/primitive/blockize_tensorize.cc
b/src/s_tir/schedule/primitive/blockize_tensorize.cc
index fe96387486..ac3c63159b 100644
--- a/src/s_tir/schedule/primitive/blockize_tensorize.cc
+++ b/src/s_tir/schedule/primitive/blockize_tensorize.cc
@@ -20,7 +20,7 @@
#include <functional>
-#include "../../../tir/transforms/simplify.h"
+#include "../../../tir/transform/simplify.h"
#include "../ir_comparator.h"
#include "../utils.h"
diff --git a/src/s_tir/schedule/primitive/cache_index.cc
b/src/s_tir/schedule/primitive/cache_index.cc
index 61e6ed8347..a9f57fe61d 100644
--- a/src/s_tir/schedule/primitive/cache_index.cc
+++ b/src/s_tir/schedule/primitive/cache_index.cc
@@ -18,8 +18,8 @@
*/
#include <tvm/arith/int_set.h>
-#include "../../../tir/transforms/common_subexpr_elim_tools.h"
-#include "../../../tir/transforms/replace_selected_expr.h"
+#include "../../../tir/transform/common_subexpr_elim_tools.h"
+#include "../../../tir/transform/replace_selected_expr.h"
#include "../utils.h"
namespace tvm {
diff --git a/src/s_tir/schedule/primitive/cache_read_write.cc
b/src/s_tir/schedule/primitive/cache_read_write.cc
index 481aec2060..c356aed3cf 100644
--- a/src/s_tir/schedule/primitive/cache_read_write.cc
+++ b/src/s_tir/schedule/primitive/cache_read_write.cc
@@ -20,7 +20,7 @@
#include <unordered_set>
#include "../../../tir/analysis/var_use_def_analysis.h"
-#include "../../../tir/transforms/ir_utils.h"
+#include "../../../tir/transform/ir_utils.h"
#include "../utils.h"
namespace tvm {
diff --git a/src/s_tir/schedule/primitive/decompose_padding.cc
b/src/s_tir/schedule/primitive/decompose_padding.cc
index c1c8b751af..e684eb4a06 100644
--- a/src/s_tir/schedule/primitive/decompose_padding.cc
+++ b/src/s_tir/schedule/primitive/decompose_padding.cc
@@ -18,7 +18,7 @@
*/
#include <tvm/ffi/reflection/registry.h>
-#include "../../../tir/transforms/ir_utils.h"
+#include "../../../tir/transform/ir_utils.h"
#include "../utils.h"
namespace tvm {
diff --git a/src/s_tir/schedule/primitive/hide_buffer_access.cc
b/src/s_tir/schedule/primitive/hide_buffer_access.cc
index a2b104ec1d..664ed1e6c9 100644
--- a/src/s_tir/schedule/primitive/hide_buffer_access.cc
+++ b/src/s_tir/schedule/primitive/hide_buffer_access.cc
@@ -16,7 +16,7 @@
* specific language governing permissions and limitations
* under the License.
*/
-#include "../../../tir/transforms/ir_utils.h"
+#include "../../../tir/transform/ir_utils.h"
#include "../utils.h"
namespace tvm {
diff --git a/src/s_tir/schedule/transform.cc b/src/s_tir/schedule/transform.cc
index a6cc76203d..f5f2356dcd 100644
--- a/src/s_tir/schedule/transform.cc
+++ b/src/s_tir/schedule/transform.cc
@@ -19,7 +19,7 @@
#include <tvm/ffi/reflection/registry.h>
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
#include "./utils.h"
namespace tvm {
diff --git a/src/s_tir/transform/canonicalize_loop.cc
b/src/s_tir/transform/canonicalize_loop.cc
index 2a0b09a7e1..5d737d7640 100644
--- a/src/s_tir/transform/canonicalize_loop.cc
+++ b/src/s_tir/transform/canonicalize_loop.cc
@@ -18,7 +18,7 @@
*/
/*!
- * \file tir/transforms/canonicalize_loop.cc
+ * \file s_tir/transform/canonicalize_loop.cc
* \brief Canonicalize all loops to start from zero and step one.
*/
#include <tvm/arith/analyzer.h>
diff --git a/src/s_tir/transform/compact_buffer_region.cc
b/src/s_tir/transform/compact_buffer_region.cc
index 1f8d0817d5..dba8e25ebe 100644
--- a/src/s_tir/transform/compact_buffer_region.cc
+++ b/src/s_tir/transform/compact_buffer_region.cc
@@ -35,7 +35,7 @@
#include "../../support/arena.h"
#include "../../support/nd_int_set.h"
#include "../../support/utils.h"
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
#include "../schedule/utils.h"
namespace tvm {
diff --git a/src/s_tir/transform/convert_blocks_to_opaque.cc
b/src/s_tir/transform/convert_blocks_to_opaque.cc
index c799cc87cb..32971922f9 100644
--- a/src/s_tir/transform/convert_blocks_to_opaque.cc
+++ b/src/s_tir/transform/convert_blocks_to_opaque.cc
@@ -26,7 +26,7 @@
#include <tvm/s_tir/transform.h>
#include <tvm/tir/stmt_functor.h>
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
namespace tvm {
namespace s_tir {
diff --git a/src/s_tir/transform/inject_double_buffer.cc
b/src/s_tir/transform/inject_double_buffer.cc
index ff77de10ae..465ccaf173 100644
--- a/src/s_tir/transform/inject_double_buffer.cc
+++ b/src/s_tir/transform/inject_double_buffer.cc
@@ -27,7 +27,7 @@
#include <tvm/tir/op.h>
#include <tvm/tir/stmt_functor.h>
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
namespace tvm {
namespace s_tir {
diff --git a/src/s_tir/transform/inject_permuted_layout.cc
b/src/s_tir/transform/inject_permuted_layout.cc
index 6483f28119..23f68f3b75 100644
--- a/src/s_tir/transform/inject_permuted_layout.cc
+++ b/src/s_tir/transform/inject_permuted_layout.cc
@@ -31,7 +31,7 @@
#include "../../arith/ir_mutator_with_analyzer.h"
#include "../../runtime/thread_storage_scope.h"
#include "../../support/utils.h"
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
namespace tvm {
namespace s_tir {
diff --git a/src/s_tir/transform/inject_software_pipeline.cc
b/src/s_tir/transform/inject_software_pipeline.cc
index 3cd2f8b8ee..33a45cdd6c 100644
--- a/src/s_tir/transform/inject_software_pipeline.cc
+++ b/src/s_tir/transform/inject_software_pipeline.cc
@@ -29,7 +29,7 @@
#include <unordered_set>
#include "../../support/utils.h"
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
#include "../schedule/utils.h"
namespace tvm {
diff --git a/src/s_tir/transform/inject_virtual_thread.cc
b/src/s_tir/transform/inject_virtual_thread.cc
index 5f305907eb..3ff7eeb7b9 100644
--- a/src/s_tir/transform/inject_virtual_thread.cc
+++ b/src/s_tir/transform/inject_virtual_thread.cc
@@ -30,7 +30,7 @@
#include <unordered_set>
#include "../../arith/ir_mutator_with_analyzer.h"
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
namespace tvm {
namespace s_tir {
diff --git a/src/s_tir/transform/lift_thread_binding.cc
b/src/s_tir/transform/lift_thread_binding.cc
index 537737b128..e0f9987c17 100644
--- a/src/s_tir/transform/lift_thread_binding.cc
+++ b/src/s_tir/transform/lift_thread_binding.cc
@@ -27,7 +27,7 @@
#include <tvm/tir/stmt_functor.h>
#include "../../runtime/thread_storage_scope.h"
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
namespace tvm {
namespace s_tir {
diff --git a/src/s_tir/transform/loop_partition.cc
b/src/s_tir/transform/loop_partition.cc
index e21f650b1f..9d2809103f 100644
--- a/src/s_tir/transform/loop_partition.cc
+++ b/src/s_tir/transform/loop_partition.cc
@@ -36,7 +36,7 @@
#include "../../arith/interval_set.h"
#include "../../runtime/thread_storage_scope.h"
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
namespace tvm {
namespace s_tir {
diff --git a/src/s_tir/transform/lower_cross_thread_reduction.cc
b/src/s_tir/transform/lower_cross_thread_reduction.cc
index 9a907c0caa..83ca4218fb 100644
--- a/src/s_tir/transform/lower_cross_thread_reduction.cc
+++ b/src/s_tir/transform/lower_cross_thread_reduction.cc
@@ -28,7 +28,7 @@
#include "../../runtime/thread_storage_scope.h"
#include "../../support/utils.h"
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
#include "../schedule/analysis.h"
namespace tvm {
diff --git a/src/s_tir/transform/lower_init_block.cc
b/src/s_tir/transform/lower_init_block.cc
index 3e5bbc1652..0efded756b 100644
--- a/src/s_tir/transform/lower_init_block.cc
+++ b/src/s_tir/transform/lower_init_block.cc
@@ -26,7 +26,7 @@
#include <tvm/tir/op.h>
#include <tvm/tir/stmt_functor.h>
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
namespace tvm {
namespace s_tir {
diff --git a/src/s_tir/transform/lower_match_buffer.cc
b/src/s_tir/transform/lower_match_buffer.cc
index 6939047bd1..5c324eecb7 100644
--- a/src/s_tir/transform/lower_match_buffer.cc
+++ b/src/s_tir/transform/lower_match_buffer.cc
@@ -30,7 +30,7 @@
#include <tvm/tir/stmt_functor.h>
#include "../../tir/ir/functor_common.h"
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
namespace tvm {
namespace s_tir {
diff --git a/src/s_tir/transform/lower_opaque_block.cc
b/src/s_tir/transform/lower_opaque_block.cc
index 39c76e9ce5..cc3ad765eb 100644
--- a/src/s_tir/transform/lower_opaque_block.cc
+++ b/src/s_tir/transform/lower_opaque_block.cc
@@ -25,7 +25,7 @@
#include <tvm/s_tir/transform.h>
#include <tvm/tir/stmt_functor.h>
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
namespace tvm {
namespace s_tir {
diff --git a/src/s_tir/transform/memhammer_lower_auto_copy.cc
b/src/s_tir/transform/memhammer_lower_auto_copy.cc
index 871f82d5e8..7e232e5ebf 100644
--- a/src/s_tir/transform/memhammer_lower_auto_copy.cc
+++ b/src/s_tir/transform/memhammer_lower_auto_copy.cc
@@ -30,7 +30,7 @@
#include <stack>
#include "../../runtime/thread_storage_scope.h"
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
#include "../schedule/utils.h"
#include "./memhammer_rewrite_rule.h"
#include "tvm/tir/stmt.h"
diff --git a/src/s_tir/transform/plan_update_buffer_allocation_location.cc
b/src/s_tir/transform/plan_update_buffer_allocation_location.cc
index 265d9574f8..a5fb66ec04 100644
--- a/src/s_tir/transform/plan_update_buffer_allocation_location.cc
+++ b/src/s_tir/transform/plan_update_buffer_allocation_location.cc
@@ -28,7 +28,7 @@
#include <tvm/tir/stmt_functor.h>
#include <tvm/tir/var.h>
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
namespace tvm {
namespace s_tir {
diff --git a/src/s_tir/transform/transform_mma_buffer_layout.cc
b/src/s_tir/transform/transform_mma_buffer_layout.cc
index 27d87f4a80..e92f24738a 100644
--- a/src/s_tir/transform/transform_mma_buffer_layout.cc
+++ b/src/s_tir/transform/transform_mma_buffer_layout.cc
@@ -24,7 +24,7 @@
#include <tvm/tir/index_map.h>
#include <tvm/tir/stmt_functor.h>
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
namespace tvm {
namespace s_tir {
diff --git a/src/s_tir/transform/unify_thread_binding.cc
b/src/s_tir/transform/unify_thread_binding.cc
index e0e6ec82ad..c43e386e71 100644
--- a/src/s_tir/transform/unify_thread_binding.cc
+++ b/src/s_tir/transform/unify_thread_binding.cc
@@ -28,7 +28,7 @@
#include <tvm/tir/stmt_functor.h>
#include "../../support/utils.h"
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
namespace tvm {
namespace s_tir {
diff --git a/src/script/printer/tir/stmt.cc b/src/script/printer/tir/stmt.cc
index 1b0774be36..9ddffe19a8 100644
--- a/src/script/printer/tir/stmt.cc
+++ b/src/script/printer/tir/stmt.cc
@@ -16,7 +16,7 @@
* specific language governing permissions and limitations
* under the License.
*/
-#include "../../../tir/transforms/ir_utils.h" // For `GetPtrStorageScope`
+#include "../../../tir/transform/ir_utils.h" // For `GetPtrStorageScope`
#include "./utils.h"
namespace tvm {
diff --git a/src/target/llvm/codegen_llvm.h b/src/target/llvm/codegen_llvm.h
index efec7ad6ad..d48e1e2f41 100644
--- a/src/target/llvm/codegen_llvm.h
+++ b/src/target/llvm/codegen_llvm.h
@@ -70,7 +70,7 @@
#include <vector>
#include "../../runtime/thread_storage_scope.h"
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
#include "codegen_params.h"
#include "llvm_instance.h"
diff --git a/src/target/source/codegen_c.h b/src/target/source/codegen_c.h
index 920e6a13a0..eae7eb7adb 100644
--- a/src/target/source/codegen_c.h
+++ b/src/target/source/codegen_c.h
@@ -39,7 +39,7 @@
#include <unordered_set>
#include <vector>
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
#include "codegen_source_base.h"
namespace tvm {
diff --git a/src/target/source/codegen_cuda.cc
b/src/target/source/codegen_cuda.cc
index 86201a2a05..650c3e3d96 100644
--- a/src/target/source/codegen_cuda.cc
+++ b/src/target/source/codegen_cuda.cc
@@ -33,7 +33,7 @@
#include <utility>
#include <vector>
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
#include "literal/cuda_half_t.h"
#include "literal/cuda_int8_t.h"
#include "ptx.h"
diff --git a/src/target/spirv/codegen_spirv.cc
b/src/target/spirv/codegen_spirv.cc
index b1d3ae9cc2..00db1ddb9b 100644
--- a/src/target/spirv/codegen_spirv.cc
+++ b/src/target/spirv/codegen_spirv.cc
@@ -31,7 +31,7 @@
#include "../../runtime/pack_args.h"
#include "../../runtime/vulkan/vulkan_common.h"
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
namespace tvm {
namespace codegen {
diff --git a/src/te/operation/create_primfunc.cc
b/src/te/operation/create_primfunc.cc
index 2b326c745a..b7f85b226b 100644
--- a/src/te/operation/create_primfunc.cc
+++ b/src/te/operation/create_primfunc.cc
@@ -38,7 +38,7 @@
#include "../../support/array.h"
#include "../../tir/ir/functor_common.h"
-#include "../../tir/transforms/ir_utils.h"
+#include "../../tir/transform/ir_utils.h"
#include "graph.h"
namespace tvm {
diff --git a/src/tir/analysis/block_access_region_detector.cc
b/src/tir/analysis/block_access_region_detector.cc
index cbd40a57a3..b79141cd1e 100644
--- a/src/tir/analysis/block_access_region_detector.cc
+++ b/src/tir/analysis/block_access_region_detector.cc
@@ -29,7 +29,7 @@
#include <unordered_map>
-#include "../transforms/ir_utils.h"
+#include "../transform/ir_utils.h"
namespace tvm {
namespace tir {
diff --git a/src/tir/analysis/verify_gpu_code.cc
b/src/tir/analysis/verify_gpu_code.cc
index e0273069cc..8b9b02e985 100644
--- a/src/tir/analysis/verify_gpu_code.cc
+++ b/src/tir/analysis/verify_gpu_code.cc
@@ -32,7 +32,7 @@
#include <tvm/tir/stmt_functor.h>
#include "../../runtime/thread_storage_scope.h"
-#include "../transforms/ir_utils.h"
+#include "../transform/ir_utils.h"
namespace tvm {
namespace tir {
diff --git a/src/tir/backend/adreno/inject_texture_alloc.cc
b/src/tir/backend/adreno/inject_texture_alloc.cc
index 4799b7ce26..f1c8297525 100644
--- a/src/tir/backend/adreno/inject_texture_alloc.cc
+++ b/src/tir/backend/adreno/inject_texture_alloc.cc
@@ -28,7 +28,7 @@
#include "../../../arith/ir_mutator_with_analyzer.h"
#include "../../../runtime/texture.h"
-#include "../../transforms/ir_utils.h"
+#include "../../transform/ir_utils.h"
namespace tvm {
namespace tir {
diff --git a/src/tir/ir/specialize.cc b/src/tir/ir/specialize.cc
index de31425770..1de16efd2a 100644
--- a/src/tir/ir/specialize.cc
+++ b/src/tir/ir/specialize.cc
@@ -30,7 +30,7 @@
#include <functional>
-#include "../transforms/ir_utils.h"
+#include "../transform/ir_utils.h"
#include "functor_common.h"
namespace tvm {
diff --git a/src/tir/transforms/annotate_device_regions.cc
b/src/tir/transform/annotate_device_regions.cc
similarity index 100%
rename from src/tir/transforms/annotate_device_regions.cc
rename to src/tir/transform/annotate_device_regions.cc
diff --git a/src/tir/transforms/arg_binder.cc b/src/tir/transform/arg_binder.cc
similarity index 100%
rename from src/tir/transforms/arg_binder.cc
rename to src/tir/transform/arg_binder.cc
diff --git a/src/tir/transforms/arg_binder.h b/src/tir/transform/arg_binder.h
similarity index 98%
rename from src/tir/transforms/arg_binder.h
rename to src/tir/transform/arg_binder.h
index fad5e4d702..e9d08fafb0 100644
--- a/src/tir/transforms/arg_binder.h
+++ b/src/tir/transform/arg_binder.h
@@ -21,8 +21,8 @@
* \file arg_binder.h
* \brief Helper utility to match and bind arguments.
*/
-#ifndef TVM_TIR_TRANSFORMS_ARG_BINDER_H_
-#define TVM_TIR_TRANSFORMS_ARG_BINDER_H_
+#ifndef TVM_TIR_TRANSFORM_ARG_BINDER_H_
+#define TVM_TIR_TRANSFORM_ARG_BINDER_H_
#include <tvm/arith/analyzer.h>
#include <tvm/tir/buffer.h>
@@ -166,4 +166,4 @@ class ArgBinder {
};
} // namespace tir
} // namespace tvm
-#endif // TVM_TIR_TRANSFORMS_ARG_BINDER_H_
+#endif // TVM_TIR_TRANSFORM_ARG_BINDER_H_
diff --git a/src/tir/transforms/bind_params.cc
b/src/tir/transform/bind_params.cc
similarity index 100%
rename from src/tir/transforms/bind_params.cc
rename to src/tir/transform/bind_params.cc
diff --git a/src/tir/transforms/bind_target.cc
b/src/tir/transform/bind_target.cc
similarity index 100%
rename from src/tir/transforms/bind_target.cc
rename to src/tir/transform/bind_target.cc
diff --git a/src/tir/transforms/bound_checker.cc
b/src/tir/transform/bound_checker.cc
similarity index 100%
rename from src/tir/transforms/bound_checker.cc
rename to src/tir/transform/bound_checker.cc
diff --git a/src/tir/transforms/combine_context_call.cc
b/src/tir/transform/combine_context_call.cc
similarity index 100%
rename from src/tir/transforms/combine_context_call.cc
rename to src/tir/transform/combine_context_call.cc
diff --git a/src/tir/transforms/common_subexpr_elim.cc
b/src/tir/transform/common_subexpr_elim.cc
similarity index 100%
rename from src/tir/transforms/common_subexpr_elim.cc
rename to src/tir/transform/common_subexpr_elim.cc
diff --git a/src/tir/transforms/common_subexpr_elim.h
b/src/tir/transform/common_subexpr_elim.h
similarity index 95%
rename from src/tir/transforms/common_subexpr_elim.h
rename to src/tir/transform/common_subexpr_elim.h
index 12a71458e1..814161cc35 100644
--- a/src/tir/transforms/common_subexpr_elim.h
+++ b/src/tir/transform/common_subexpr_elim.h
@@ -25,8 +25,8 @@
places where the expression was used are replaced with the freshly
introduced variable.
*/
-#ifndef TVM_TIR_TRANSFORMS_COMMON_SUBEXPR_ELIM_H_
-#define TVM_TIR_TRANSFORMS_COMMON_SUBEXPR_ELIM_H_
+#ifndef TVM_TIR_TRANSFORM_COMMON_SUBEXPR_ELIM_H_
+#define TVM_TIR_TRANSFORM_COMMON_SUBEXPR_ELIM_H_
#include <tvm/tir/expr.h>
#include <tvm/tir/expr_functor.h>
@@ -91,4 +91,4 @@ class CommonSubexpressionEliminator : public StmtExprMutator {
} // namespace tir
} // namespace tvm
-#endif // TVM_TIR_TRANSFORMS_COMMON_SUBEXPR_ELIM_H_
+#endif // TVM_TIR_TRANSFORM_COMMON_SUBEXPR_ELIM_H_
diff --git a/src/tir/transforms/common_subexpr_elim_tools.cc
b/src/tir/transform/common_subexpr_elim_tools.cc
similarity index 100%
rename from src/tir/transforms/common_subexpr_elim_tools.cc
rename to src/tir/transform/common_subexpr_elim_tools.cc
diff --git a/src/tir/transforms/common_subexpr_elim_tools.h
b/src/tir/transform/common_subexpr_elim_tools.h
similarity index 98%
rename from src/tir/transforms/common_subexpr_elim_tools.h
rename to src/tir/transform/common_subexpr_elim_tools.h
index ab1e76592a..b9c056dcf2 100644
--- a/src/tir/transforms/common_subexpr_elim_tools.h
+++ b/src/tir/transform/common_subexpr_elim_tools.h
@@ -23,8 +23,8 @@
by the Common Subexpression Elimination (CSE) pass.
*/
-#ifndef TVM_TIR_TRANSFORMS_COMMON_SUBEXPR_ELIM_TOOLS_H_
-#define TVM_TIR_TRANSFORMS_COMMON_SUBEXPR_ELIM_TOOLS_H_
+#ifndef TVM_TIR_TRANSFORM_COMMON_SUBEXPR_ELIM_TOOLS_H_
+#define TVM_TIR_TRANSFORM_COMMON_SUBEXPR_ELIM_TOOLS_H_
#include <tvm/ffi/string.h>
#include <tvm/tir/analysis.h> // For the ExprDeepEqual analysis
@@ -220,4 +220,4 @@ void
InsertVectorToSortedSemanticComputations(std::vector<std::pair<PrimExpr, si
} // namespace tir
} // namespace tvm
-#endif // TVM_TIR_TRANSFORMS_COMMON_SUBEXPR_ELIM_TOOLS_H_
+#endif // TVM_TIR_TRANSFORM_COMMON_SUBEXPR_ELIM_TOOLS_H_
diff --git a/src/tir/transforms/convert_for_loops_serial.cc
b/src/tir/transform/convert_for_loops_serial.cc
similarity index 97%
rename from src/tir/transforms/convert_for_loops_serial.cc
rename to src/tir/transform/convert_for_loops_serial.cc
index 691d8b885c..c695d8b3f7 100644
--- a/src/tir/transforms/convert_for_loops_serial.cc
+++ b/src/tir/transform/convert_for_loops_serial.cc
@@ -18,7 +18,7 @@
*/
/*!
- * \file tir/transforms/convert_for_loops_serial.cc
+ * \file tir/transform/convert_for_loops_serial.cc
* \brief Convert all for loops to serial for lesser memory consumption
*/
#include <tvm/arith/analyzer.h>
diff --git a/src/tir/transforms/decorate_device_scope.cc
b/src/tir/transform/decorate_device_scope.cc
similarity index 100%
rename from src/tir/transforms/decorate_device_scope.cc
rename to src/tir/transform/decorate_device_scope.cc
diff --git a/src/tir/transforms/default_gpu_schedule.cc
b/src/tir/transform/default_gpu_schedule.cc
similarity index 100%
rename from src/tir/transforms/default_gpu_schedule.cc
rename to src/tir/transform/default_gpu_schedule.cc
diff --git a/src/tir/transforms/dtype_conversion.cc
b/src/tir/transform/dtype_conversion.cc
similarity index 100%
rename from src/tir/transforms/dtype_conversion.cc
rename to src/tir/transform/dtype_conversion.cc
diff --git a/src/tir/transforms/dtype_conversion.h
b/src/tir/transform/dtype_conversion.h
similarity index 98%
rename from src/tir/transforms/dtype_conversion.h
rename to src/tir/transform/dtype_conversion.h
index bd6f34b3ac..310ed21c76 100644
--- a/src/tir/transforms/dtype_conversion.h
+++ b/src/tir/transform/dtype_conversion.h
@@ -21,8 +21,8 @@
* \file dtype_conversion.h
* \brief Header file of data type conversion routines.
*/
-#ifndef TVM_TIR_TRANSFORMS_DTYPE_CONVERSION_H_
-#define TVM_TIR_TRANSFORMS_DTYPE_CONVERSION_H_
+#ifndef TVM_TIR_TRANSFORM_DTYPE_CONVERSION_H_
+#define TVM_TIR_TRANSFORM_DTYPE_CONVERSION_H_
#include <tvm/tir/builtin.h>
#include <tvm/tir/op.h>
@@ -198,4 +198,4 @@ PrimExpr DTypeConversion(PrimExpr src_value, DataType
tgt_dtype,
} // namespace tir
} // namespace tvm
-#endif // TVM_TIR_TRANSFORMS_DTYPE_CONVERSION_H_
+#endif // TVM_TIR_TRANSFORM_DTYPE_CONVERSION_H_
diff --git a/src/tir/transforms/extract_constants.cc
b/src/tir/transform/extract_constants.cc
similarity index 100%
rename from src/tir/transforms/extract_constants.cc
rename to src/tir/transform/extract_constants.cc
diff --git a/src/tir/transforms/flatten_buffer.cc
b/src/tir/transform/flatten_buffer.cc
similarity index 100%
rename from src/tir/transforms/flatten_buffer.cc
rename to src/tir/transform/flatten_buffer.cc
diff --git a/src/tir/transforms/force_narrow_index_to_i32.cc
b/src/tir/transform/force_narrow_index_to_i32.cc
similarity index 100%
rename from src/tir/transforms/force_narrow_index_to_i32.cc
rename to src/tir/transform/force_narrow_index_to_i32.cc
diff --git a/src/tir/transforms/hoist_expression.cc
b/src/tir/transform/hoist_expression.cc
similarity index 100%
rename from src/tir/transforms/hoist_expression.cc
rename to src/tir/transform/hoist_expression.cc
diff --git a/src/tir/transforms/inject_ptx_async_copy.cc
b/src/tir/transform/inject_ptx_async_copy.cc
similarity index 100%
rename from src/tir/transforms/inject_ptx_async_copy.cc
rename to src/tir/transform/inject_ptx_async_copy.cc
diff --git a/src/tir/transforms/inject_ptx_ldg32.cc
b/src/tir/transform/inject_ptx_ldg32.cc
similarity index 100%
rename from src/tir/transforms/inject_ptx_ldg32.cc
rename to src/tir/transform/inject_ptx_ldg32.cc
diff --git a/src/tir/transforms/inject_rolling_buffer.cc
b/src/tir/transform/inject_rolling_buffer.cc
similarity index 100%
rename from src/tir/transforms/inject_rolling_buffer.cc
rename to src/tir/transform/inject_rolling_buffer.cc
diff --git a/src/tir/transforms/inline_private_functions.cc
b/src/tir/transform/inline_private_functions.cc
similarity index 100%
rename from src/tir/transforms/inline_private_functions.cc
rename to src/tir/transform/inline_private_functions.cc
diff --git a/src/tir/transforms/ir_utils.cc b/src/tir/transform/ir_utils.cc
similarity index 100%
rename from src/tir/transforms/ir_utils.cc
rename to src/tir/transform/ir_utils.cc
diff --git a/src/tir/transforms/ir_utils.h b/src/tir/transform/ir_utils.h
similarity index 99%
rename from src/tir/transforms/ir_utils.h
rename to src/tir/transform/ir_utils.h
index fdf4def699..0616e23e7e 100644
--- a/src/tir/transforms/ir_utils.h
+++ b/src/tir/transform/ir_utils.h
@@ -21,8 +21,8 @@
* \file ir_utils.h
* \brief Helper functions to construct and compose IR nodes.
*/
-#ifndef TVM_TIR_TRANSFORMS_IR_UTILS_H_
-#define TVM_TIR_TRANSFORMS_IR_UTILS_H_
+#ifndef TVM_TIR_TRANSFORM_IR_UTILS_H_
+#define TVM_TIR_TRANSFORM_IR_UTILS_H_
#include <tvm/arith/int_set.h>
#include <tvm/arith/int_solver.h>
@@ -356,4 +356,4 @@ std::optional<bool> IsHostFunc(const PrimFunc& func);
} // namespace tir
} // namespace tvm
-#endif // TVM_TIR_TRANSFORMS_IR_UTILS_H_
+#endif // TVM_TIR_TRANSFORM_IR_UTILS_H_
diff --git a/src/tir/transforms/lower_async_dma.cc
b/src/tir/transform/lower_async_dma.cc
similarity index 100%
rename from src/tir/transforms/lower_async_dma.cc
rename to src/tir/transform/lower_async_dma.cc
diff --git a/src/tir/transforms/lower_custom_datatypes.cc
b/src/tir/transform/lower_custom_datatypes.cc
similarity index 100%
rename from src/tir/transforms/lower_custom_datatypes.cc
rename to src/tir/transform/lower_custom_datatypes.cc
diff --git a/src/tir/transforms/lower_device_kernel_launch.cc
b/src/tir/transform/lower_device_kernel_launch.cc
similarity index 100%
rename from src/tir/transforms/lower_device_kernel_launch.cc
rename to src/tir/transform/lower_device_kernel_launch.cc
diff --git a/src/tir/transforms/lower_device_storage_access_info.cc
b/src/tir/transform/lower_device_storage_access_info.cc
similarity index 100%
rename from src/tir/transforms/lower_device_storage_access_info.cc
rename to src/tir/transform/lower_device_storage_access_info.cc
diff --git a/src/tir/transforms/lower_intrin.cc
b/src/tir/transform/lower_intrin.cc
similarity index 100%
rename from src/tir/transforms/lower_intrin.cc
rename to src/tir/transform/lower_intrin.cc
diff --git a/src/tir/transforms/lower_thread_allreduce.cc
b/src/tir/transform/lower_thread_allreduce.cc
similarity index 100%
rename from src/tir/transforms/lower_thread_allreduce.cc
rename to src/tir/transform/lower_thread_allreduce.cc
diff --git a/src/tir/transforms/lower_tvm_builtin.cc
b/src/tir/transform/lower_tvm_builtin.cc
similarity index 99%
rename from src/tir/transforms/lower_tvm_builtin.cc
rename to src/tir/transform/lower_tvm_builtin.cc
index 66e13791f3..4add282272 100644
--- a/src/tir/transforms/lower_tvm_builtin.cc
+++ b/src/tir/transform/lower_tvm_builtin.cc
@@ -19,7 +19,7 @@
/*!
* Lower TVM related builtin intrinsics such as packed call.
- * \file tir/transforms/lower_tvm_buildin.cc
+ * \file tir/transform/lower_tvm_builtin.cc
*/
#include <tvm/ffi/function.h>
#include <tvm/ffi/reflection/registry.h>
diff --git a/src/tir/transforms/lower_vtcm_alloc.cc
b/src/tir/transform/lower_vtcm_alloc.cc
similarity index 100%
rename from src/tir/transforms/lower_vtcm_alloc.cc
rename to src/tir/transform/lower_vtcm_alloc.cc
diff --git a/src/tir/transforms/lower_warp_memory.cc
b/src/tir/transform/lower_warp_memory.cc
similarity index 100%
rename from src/tir/transforms/lower_warp_memory.cc
rename to src/tir/transform/lower_warp_memory.cc
diff --git a/src/tir/transforms/make_packed_api.cc
b/src/tir/transform/make_packed_api.cc
similarity index 100%
rename from src/tir/transforms/make_packed_api.cc
rename to src/tir/transform/make_packed_api.cc
diff --git a/src/tir/transforms/make_unpacked_api.cc
b/src/tir/transform/make_unpacked_api.cc
similarity index 100%
rename from src/tir/transforms/make_unpacked_api.cc
rename to src/tir/transform/make_unpacked_api.cc
diff --git a/src/tir/transforms/merge_shared_memory_allocations.cc
b/src/tir/transform/merge_shared_memory_allocations.cc
similarity index 100%
rename from src/tir/transforms/merge_shared_memory_allocations.cc
rename to src/tir/transform/merge_shared_memory_allocations.cc
diff --git a/src/tir/transforms/narrow_datatype.cc
b/src/tir/transform/narrow_datatype.cc
similarity index 100%
rename from src/tir/transforms/narrow_datatype.cc
rename to src/tir/transform/narrow_datatype.cc
diff --git a/src/tir/transforms/primfunc_utils.cc
b/src/tir/transform/primfunc_utils.cc
similarity index 100%
rename from src/tir/transforms/primfunc_utils.cc
rename to src/tir/transform/primfunc_utils.cc
diff --git a/src/tir/transforms/profile_instrumentation.cc
b/src/tir/transform/profile_instrumentation.cc
similarity index 100%
rename from src/tir/transforms/profile_instrumentation.cc
rename to src/tir/transform/profile_instrumentation.cc
diff --git a/src/tir/transforms/reduce_branching_through_overcompute.cc
b/src/tir/transform/reduce_branching_through_overcompute.cc
similarity index 100%
rename from src/tir/transforms/reduce_branching_through_overcompute.cc
rename to src/tir/transform/reduce_branching_through_overcompute.cc
diff --git a/src/tir/transforms/remap_thread_axis.cc
b/src/tir/transform/remap_thread_axis.cc
similarity index 100%
rename from src/tir/transforms/remap_thread_axis.cc
rename to src/tir/transform/remap_thread_axis.cc
diff --git a/src/tir/transforms/remove_assume.cc
b/src/tir/transform/remove_assume.cc
similarity index 100%
rename from src/tir/transforms/remove_assume.cc
rename to src/tir/transform/remove_assume.cc
diff --git a/src/tir/transforms/remove_no_op.cc
b/src/tir/transform/remove_no_op.cc
similarity index 100%
rename from src/tir/transforms/remove_no_op.cc
rename to src/tir/transform/remove_no_op.cc
diff --git a/src/tir/transforms/remove_no_op.h
b/src/tir/transform/remove_no_op.h
similarity index 93%
rename from src/tir/transforms/remove_no_op.h
rename to src/tir/transform/remove_no_op.h
index e24c32b5da..493b98c90d 100644
--- a/src/tir/transforms/remove_no_op.h
+++ b/src/tir/transform/remove_no_op.h
@@ -21,8 +21,8 @@
* \file remove_no_op.h
* \brief Helper functions to construct and compose IR nodes.
*/
-#ifndef TVM_TIR_TRANSFORMS_REMOVE_NO_OP_H_
-#define TVM_TIR_TRANSFORMS_REMOVE_NO_OP_H_
+#ifndef TVM_TIR_TRANSFORM_REMOVE_NO_OP_H_
+#define TVM_TIR_TRANSFORM_REMOVE_NO_OP_H_
#include <tvm/arith/analyzer.h>
#include <tvm/tir/stmt.h>
@@ -57,4 +57,4 @@ Stmt RemoveNoOp(Stmt stmt, arith::Analyzer* analyzer,
} // namespace tir
} // namespace tvm
-#endif // TVM_TIR_TRANSFORMS_REMOVE_NO_OP_H_
+#endif // TVM_TIR_TRANSFORM_REMOVE_NO_OP_H_
diff --git a/src/tir/transforms/remove_store_undef.cc
b/src/tir/transform/remove_store_undef.cc
similarity index 100%
rename from src/tir/transforms/remove_store_undef.cc
rename to src/tir/transform/remove_store_undef.cc
diff --git a/src/tir/transforms/remove_weight_layout_rewrite_block.cc
b/src/tir/transform/remove_weight_layout_rewrite_block.cc
similarity index 100%
rename from src/tir/transforms/remove_weight_layout_rewrite_block.cc
rename to src/tir/transform/remove_weight_layout_rewrite_block.cc
diff --git a/src/tir/transforms/renew_defs.cc b/src/tir/transform/renew_defs.cc
similarity index 100%
rename from src/tir/transforms/renew_defs.cc
rename to src/tir/transform/renew_defs.cc
diff --git a/src/tir/transforms/renormalize_split_pattern.cc
b/src/tir/transform/renormalize_split_pattern.cc
similarity index 100%
rename from src/tir/transforms/renormalize_split_pattern.cc
rename to src/tir/transform/renormalize_split_pattern.cc
diff --git a/src/tir/transforms/replace_global_vars.cc
b/src/tir/transform/replace_global_vars.cc
similarity index 98%
rename from src/tir/transforms/replace_global_vars.cc
rename to src/tir/transform/replace_global_vars.cc
index b16926056b..8bb7dee378 100644
--- a/src/tir/transforms/replace_global_vars.cc
+++ b/src/tir/transform/replace_global_vars.cc
@@ -19,7 +19,7 @@
/*!
*
- * \file src/tir/transforms/replace_global_vars.cc
+ * \file src/tir/transform/replace_global_vars.cc
*
* \brief GlobalVar replacement across IR types
*/
diff --git a/src/tir/transforms/replace_selected_expr.cc
b/src/tir/transform/replace_selected_expr.cc
similarity index 100%
rename from src/tir/transforms/replace_selected_expr.cc
rename to src/tir/transform/replace_selected_expr.cc
diff --git a/src/tir/transforms/replace_selected_expr.h
b/src/tir/transform/replace_selected_expr.h
similarity index 94%
rename from src/tir/transforms/replace_selected_expr.h
rename to src/tir/transform/replace_selected_expr.h
index 925615726e..e3ee62fb8d 100644
--- a/src/tir/transforms/replace_selected_expr.h
+++ b/src/tir/transform/replace_selected_expr.h
@@ -24,8 +24,8 @@
with a predicate by another expression.
*/
-#ifndef TVM_TIR_TRANSFORMS_REPLACE_SELECTED_EXPR_H_
-#define TVM_TIR_TRANSFORMS_REPLACE_SELECTED_EXPR_H_
+#ifndef TVM_TIR_TRANSFORM_REPLACE_SELECTED_EXPR_H_
+#define TVM_TIR_TRANSFORM_REPLACE_SELECTED_EXPR_H_
#include <tvm/tir/expr.h>
#include <tvm/tir/expr_functor.h>
@@ -72,4 +72,4 @@ class ReplaceSelectedExpr : public StmtExprMutator {
} // namespace tir
} // namespace tvm
-#endif // TVM_TIR_TRANSFORMS_REPLACE_SELECTED_EXPR_H_
+#endif // TVM_TIR_TRANSFORM_REPLACE_SELECTED_EXPR_H_
diff --git a/src/tir/transforms/rewrite_unsafe_select.cc
b/src/tir/transform/rewrite_unsafe_select.cc
similarity index 100%
rename from src/tir/transforms/rewrite_unsafe_select.cc
rename to src/tir/transform/rewrite_unsafe_select.cc
diff --git a/src/tir/transforms/simplify.cc b/src/tir/transform/simplify.cc
similarity index 99%
rename from src/tir/transforms/simplify.cc
rename to src/tir/transform/simplify.cc
index a3365db9b7..648710584a 100644
--- a/src/tir/transforms/simplify.cc
+++ b/src/tir/transform/simplify.cc
@@ -22,7 +22,7 @@
* \brief Statement simplifier based on analyzer
*/
-#include "../../tir/transforms/simplify.h"
+#include "../../tir/transform/simplify.h"
#include <tvm/arith/analyzer.h>
#include <tvm/ffi/function.h>
diff --git a/src/tir/transforms/simplify.h b/src/tir/transform/simplify.h
similarity index 90%
rename from src/tir/transforms/simplify.h
rename to src/tir/transform/simplify.h
index 25c9dd5791..03a493d51b 100644
--- a/src/tir/transforms/simplify.h
+++ b/src/tir/transform/simplify.h
@@ -21,8 +21,8 @@
* \file simplify.h
* \brief Helper functions to construct and compose IR nodes.
*/
-#ifndef TVM_TIR_TRANSFORMS_SIMPLIFY_H_
-#define TVM_TIR_TRANSFORMS_SIMPLIFY_H_
+#ifndef TVM_TIR_TRANSFORM_SIMPLIFY_H_
+#define TVM_TIR_TRANSFORM_SIMPLIFY_H_
#include <tvm/arith/analyzer.h>
#include <tvm/tir/function.h>
@@ -38,4 +38,4 @@ PrimFunc Simplify(PrimFunc stmt, arith::Analyzer* analyzer);
} // namespace tir
} // namespace tvm
-#endif // TVM_TIR_TRANSFORMS_SIMPLIFY_H_
+#endif // TVM_TIR_TRANSFORM_SIMPLIFY_H_
diff --git a/src/tir/transforms/skip_assert.cc
b/src/tir/transform/skip_assert.cc
similarity index 100%
rename from src/tir/transforms/skip_assert.cc
rename to src/tir/transform/skip_assert.cc
diff --git a/src/tir/transforms/split_host_device.cc
b/src/tir/transform/split_host_device.cc
similarity index 100%
rename from src/tir/transforms/split_host_device.cc
rename to src/tir/transform/split_host_device.cc
diff --git a/src/tir/transforms/storage_access.cc
b/src/tir/transform/storage_access.cc
similarity index 100%
rename from src/tir/transforms/storage_access.cc
rename to src/tir/transform/storage_access.cc
diff --git a/src/tir/transforms/storage_access.h
b/src/tir/transform/storage_access.h
similarity index 97%
rename from src/tir/transforms/storage_access.h
rename to src/tir/transform/storage_access.h
index 10b26f7c2a..7c96068229 100644
--- a/src/tir/transforms/storage_access.h
+++ b/src/tir/transform/storage_access.h
@@ -21,8 +21,8 @@
* \file storage_access.h
* \brief Common data structure for storage access analysis.
*/
-#ifndef TVM_TIR_TRANSFORMS_STORAGE_ACCESS_H_
-#define TVM_TIR_TRANSFORMS_STORAGE_ACCESS_H_
+#ifndef TVM_TIR_TRANSFORM_STORAGE_ACCESS_H_
+#define TVM_TIR_TRANSFORM_STORAGE_ACCESS_H_
#include <tvm/arith/int_set.h>
#include <tvm/ir/attrs.h>
@@ -142,4 +142,4 @@ class StorageAccessVisitor : public StmtExprVisitor {
};
} // namespace tir
} // namespace tvm
-#endif // TVM_TIR_TRANSFORMS_STORAGE_ACCESS_H_
+#endif // TVM_TIR_TRANSFORM_STORAGE_ACCESS_H_
diff --git a/src/tir/transforms/storage_rewrite.cc
b/src/tir/transform/storage_rewrite.cc
similarity index 100%
rename from src/tir/transforms/storage_rewrite.cc
rename to src/tir/transform/storage_rewrite.cc
diff --git a/src/tir/transforms/tensorcore_infer_fragment.cc
b/src/tir/transform/tensorcore_infer_fragment.cc
similarity index 100%
rename from src/tir/transforms/tensorcore_infer_fragment.cc
rename to src/tir/transform/tensorcore_infer_fragment.cc
diff --git a/src/tir/transforms/thread_storage_sync.cc
b/src/tir/transform/thread_storage_sync.cc
similarity index 100%
rename from src/tir/transforms/thread_storage_sync.cc
rename to src/tir/transform/thread_storage_sync.cc
diff --git a/src/tir/transforms/unroll_loop.cc
b/src/tir/transform/unroll_loop.cc
similarity index 100%
rename from src/tir/transforms/unroll_loop.cc
rename to src/tir/transform/unroll_loop.cc
diff --git a/src/tir/transforms/unsupported_dtype_legalize.cc
b/src/tir/transform/unsupported_dtype_legalize.cc
similarity index 100%
rename from src/tir/transforms/unsupported_dtype_legalize.cc
rename to src/tir/transform/unsupported_dtype_legalize.cc
diff --git a/src/tir/transforms/update_pointer_storage_scope.cc
b/src/tir/transform/update_pointer_storage_scope.cc
similarity index 100%
rename from src/tir/transforms/update_pointer_storage_scope.cc
rename to src/tir/transform/update_pointer_storage_scope.cc
diff --git a/src/tir/transforms/update_pointer_storage_scope.h
b/src/tir/transform/update_pointer_storage_scope.h
similarity index 90%
rename from src/tir/transforms/update_pointer_storage_scope.h
rename to src/tir/transform/update_pointer_storage_scope.h
index a2f7027ce4..1579882d86 100644
--- a/src/tir/transforms/update_pointer_storage_scope.h
+++ b/src/tir/transform/update_pointer_storage_scope.h
@@ -21,8 +21,8 @@
* \file update_pointer_storage_scope.h
* \brief A pass to update storage scopes for buffer variables.
*/
-#ifndef TVM_TIR_TRANSFORMS_UPDATE_POINTER_STORAGE_SCOPE_H_
-#define TVM_TIR_TRANSFORMS_UPDATE_POINTER_STORAGE_SCOPE_H_
+#ifndef TVM_TIR_TRANSFORM_UPDATE_POINTER_STORAGE_SCOPE_H_
+#define TVM_TIR_TRANSFORM_UPDATE_POINTER_STORAGE_SCOPE_H_
#include <tvm/tir/expr.h>
#include <tvm/tir/op.h>
@@ -56,4 +56,4 @@ class UpdatePointerStorageScope : public StmtExprMutator {
} // namespace tir
} // namespace tvm
-#endif // TVM_TIR_TRANSFORMS_UPDATE_POINTER_STORAGE_SCOPE_H_
+#endif // TVM_TIR_TRANSFORM_UPDATE_POINTER_STORAGE_SCOPE_H_
diff --git a/src/tir/transforms/using_assume_to_reduce_branches.cc
b/src/tir/transform/using_assume_to_reduce_branches.cc
similarity index 100%
rename from src/tir/transforms/using_assume_to_reduce_branches.cc
rename to src/tir/transform/using_assume_to_reduce_branches.cc
diff --git a/src/tir/transforms/vectorize_loop.cc
b/src/tir/transform/vectorize_loop.cc
similarity index 100%
rename from src/tir/transforms/vectorize_loop.cc
rename to src/tir/transform/vectorize_loop.cc
