[REFACTOR][S-TIR] Migrate tir/schedule to s_tir

[tqchen]

This PR lifts the previous tir/schedule to s_tir namespace.

[gemini-code-assist]

Summary of Changes

Hello @tqchen, I'm Gemini Code Assist¹! I'm currently reviewing this pull request and will post my feedback shortly. In the meantime, here's a summary to help you and other reviewers quickly get up to speed!

This pull request undertakes a significant refactoring of the TensorIR (TIR) scheduling infrastructure. The primary goal is to establish a dedicated s_tir namespace and directory structure for these components, enhancing modularity and organization. The changes involve updating all relevant C++ include paths, namespace declarations, and type references, alongside modifications to Python bindings and global function calls to align with the new naming convention.

Highlights

• Namespace Migration: The core TensorIR (TIR) scheduling components have been migrated from the tir/schedule namespace to a new s_tir namespace, affecting numerous C++ and Python files.
• Directory Restructuring: The corresponding file paths for the migrated components have been updated, moving files from include/tvm/tir/schedule/ and src/tir/schedule/ to include/tvm/s_tir/schedule/ and src/s_tir/schedule/ respectively.
• FFI and Type Reference Updates: All references to tir::Schedule, tir::Trace, tir::LoopRV, tir::SBlockRV, and related types, as well as FFI registration names (e.g., tir.schedule.X to s_tir.schedule.X), have been updated to reflect the new s_tir namespace.

🧠 New Feature in Public Preview: You can now enable Memory to help Gemini Code Assist learn from your team's feedback. This makes future code reviews more consistent and personalized to your project's style. Click here to enable Memory in your admin console.

Changelog

• include/tvm/meta_schedule/cost_model.h
  • Updated include path for schedule header from tvm/tir/schedule/schedule.h to tvm/s_tir/schedule/schedule.h.
• include/tvm/meta_schedule/database.h
  • Updated include paths for schedule and trace headers to tvm/s_tir/schedule/schedule.h and tvm/s_tir/schedule/trace.h.
  • Changed type references from tir::Trace to s_tir::Trace in TuningRecordNode.
  • Updated constructor and method signatures to use s_tir::Trace and s_tir::Schedule.
• include/tvm/meta_schedule/measure_candidate.h
  • Updated include path for schedule header to tvm/s_tir/schedule/schedule.h.
  • Changed type references from tir::Schedule to s_tir::Schedule in MeasureCandidateNode and its constructor.
• include/tvm/meta_schedule/mutator.h
  • Updated include paths for schedule and trace headers to tvm/s_tir/schedule/schedule.h and tvm/s_tir/schedule/trace.h.
  • Changed type references from tir::Trace to s_tir::Trace in Apply method signatures and FApply function type.
• include/tvm/meta_schedule/postproc.h
  • Updated include path for schedule header to tvm/s_tir/schedule/schedule.h.
  • Changed type references from tir::Schedule to s_tir::Schedule in Apply method signatures and FApply function type.
• include/tvm/meta_schedule/schedule/cuda/thread_bind.h
  • Updated include path for schedule header to tvm/s_tir/schedule/schedule.h.
  • Changed type references from tir::ExprRV, tir::LoopRV, tir::SBlockRV, and tir::Schedule to their s_tir counterparts in function signatures.
• include/tvm/meta_schedule/schedule/generic/winograd.h
  • Updated include path for schedule header to tvm/s_tir/schedule/schedule.h.
  • Changed type references from tir::SBlockRV and tir::Schedule to their s_tir counterparts in function signatures.
• include/tvm/meta_schedule/schedule_rule.h
  • Updated include path for schedule header to tvm/s_tir/schedule/schedule.h.
  • Changed type references from tir::Schedule and tir::SBlockRV to their s_tir counterparts in Apply method signatures and FApply function type.
• include/tvm/meta_schedule/search_strategy.h
  • Updated include path for schedule header to tvm/s_tir/schedule/schedule.h.
  • Changed type references from tir::Schedule to s_tir::Schedule in PreTuning method signatures and FPreTuning function type.
• include/tvm/meta_schedule/space_generator.h
  • Updated include path for schedule header to tvm/s_tir/schedule/schedule.h.
  • Changed type references from tir::Schedule to s_tir::Schedule in GenerateDesignSpace method signatures and FGenerateDesignSpace function type.
• include/tvm/tir/schedule/instruction.h
  • File renamed to include/tvm/s_tir/schedule/instruction.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated TVM_FFI_DECLARE_OBJECT_INFO_FINAL and TVM_INST_KIND_REGISTER_VAR_DEF macros to use s_tir.
• include/tvm/tir/schedule/schedule.h
  • File renamed to include/tvm/s_tir/schedule/schedule.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated TVM_FFI_DECLARE_OBJECT_INFO_FINAL macros to use s_tir.
  • Updated include paths for state.h and trace.h to tvm/s_tir/schedule/state.h and tvm/s_tir/schedule/trace.h.
• include/tvm/tir/schedule/state.h
  • File renamed to include/tvm/s_tir/schedule/state.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated TVM_FFI_DECLARE_OBJECT_INFO_FINAL macros to use s_tir.
• include/tvm/tir/schedule/trace.h
  • File renamed to include/tvm/s_tir/schedule/trace.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated TVM_FFI_DECLARE_OBJECT_INFO_FINAL macros to use s_tir.
  • Updated include path for instruction.h to tvm/s_tir/schedule/instruction.h.
• python/tvm/dlight/analysis/common_analysis.py
  • Updated global function calls from tir.schedule.GetSBlockRealize, tir.schedule.HasIfThenElse, and tir.schedule.NormalizePrimFunc to their s_tir.schedule counterparts.
• python/tvm/s_tir/schedule/_ffi_api.py
  • Updated tvm_ffi.init_ffi_api from "tir.schedule" to "s_tir.schedule".
• python/tvm/s_tir/schedule/analysis.py
  • Updated @tvm_ffi.register_object decorators from "tir.schedule.TensorizeInfo" and "tir.schedule.AutoTensorizeMappingInfo" to their s_tir.schedule counterparts.
• python/tvm/s_tir/schedule/instruction.py
  • Updated @_register_object decorators from "tir.InstructionKind" and "tir.Instruction" to their s_tir counterparts.
• python/tvm/s_tir/schedule/schedule.py
  • Updated @_register_object decorators from "tir.LoopRV", "tir.SBlockRV", and "tir.Schedule" to their s_tir counterparts.
• python/tvm/s_tir/schedule/state.py
  • Updated @register_object decorator from "tir.ScheduleState" to "s_tir.ScheduleState".
• python/tvm/s_tir/schedule/trace.py
  • Updated @_register_object decorator from "tir.Trace" to "s_tir.Trace".
• python/tvm/tir/analysis/analysis.py
  • Updated global function call from tvm.ffi.get_global_func("tir.schedule.HasIfThenElse") to tvm.ffi.get_global_func("s_tir.schedule.HasIfThenElse").
• src/meta_schedule/database/database.cc
  • Updated type references from tir::Trace to s_tir::Trace in TuningRecord constructor and TuningRecord::FromJSON.
  • Updated type references from tir::Schedule to s_tir::Schedule in TuningRecordNode::AsMeasureCandidate and DatabaseNode::QuerySchedule.
  • Updated lambda function signature in TVM_FFI_STATIC_INIT_BLOCK to use s_tir::Trace.
• src/meta_schedule/database/schedule_fn_database.cc
  • Updated type references from tir::Schedule to s_tir::Schedule in ScheduleFnDatabaseNode member, QueryTuningRecord, QuerySchedule methods, and Database::ScheduleFnDatabase function.
• src/meta_schedule/feature_extractor/per_store_feature.cc
  • Changed namespace from tvm::tir to tvm::s_tir.
  • Updated PassListForPerStoreFeature to use s_tir::transform::SimplifyForFeatureExtraction.
  • Changed type references from tir::Feature to s_tir::Feature and tir::groupX::Feature to s_tir::groupX::Feature.
• src/meta_schedule/mutator/mutate_compute_location.cc
  • Updated using directives for Instruction, InstructionKind, Trace to s_tir namespace.
  • Changed type references from tir::Schedule to s_tir::Schedule and tir::SBlockRV to s_tir::SBlockRV.
  • Updated sampling function calls from tir::SampleInt to s_tir::SampleInt.
• src/meta_schedule/mutator/mutate_parallel.cc
  • Changed namespace from tvm::tir to tvm::s_tir.
  • Updated using directives for Instruction, Trace to s_tir namespace.
  • Changed attr::meta_schedule_parallel to tir::attr::meta_schedule_parallel.
  • Updated global function calls like tir::GetSBlocks, tir::IsAnnotateWithParallel, tir::GetInstGetSBlockOutput, tir::AnalyzeParallel, tir::GetNumFusedLoops, tir::SampleInt, tir::ReplaceAnnValue to use s_tir namespace.
• src/meta_schedule/mutator/mutate_thread_binding.cc
  • Updated using directives for Instruction, InstructionKind, Trace to s_tir namespace.
  • Changed type references from tir::InstructionNode to s_tir::InstructionNode and tir::LoopRVNode to s_tir::LoopRVNode.
  • Updated sampling function calls from tir::SampleInt and tir::MakeMultinomialSampler to s_tir::SampleInt and s_tir::MakeMultinomialSampler.
• src/meta_schedule/mutator/mutate_tile_size.cc
  • Updated using directives for Instruction, InstructionKind, Trace to s_tir namespace.
  • Changed type references from tir::ExprRVNode to s_tir::ExprRVNode.
  • Updated sampling function calls from tir::SampleInt and tir::MakeMultinomialSampler to s_tir::SampleInt and s_tir::MakeMultinomialSampler.
• src/meta_schedule/mutator/mutate_unroll.cc
  • Changed namespace from tvm::tir to tvm::s_tir.
  • Updated using directives for Instruction, Trace to s_tir namespace.
  • Changed attr::meta_schedule_unroll_explicit and attr::meta_schedule_unroll_implicit to tir::attr::meta_schedule_unroll_explicit and tir::attr::meta_schedule_unroll_implicit.
  • Updated sampling function calls from tir::SampleInt and tir::MakeMultinomialSampler to s_tir::SampleInt and s_tir::MakeMultinomialSampler.
• src/meta_schedule/mutator/mutator.cc
  • Updated type references from tir::Trace to s_tir::Trace in PyMutatorNode::Apply and lambda function in TVM_FFI_STATIC_INIT_BLOCK.
• src/meta_schedule/postproc/disallow_async_strided_mem_copy.cc
  • Changed namespace from tvm::tir to tvm::s_tir.
  • Updated type references from tir::Schedule to s_tir::Schedule in Apply method.
  • Changed tir::AsyncStridedMemCopyFinder::Find to s_tir::AsyncStridedMemCopyFinder::Find.
• src/meta_schedule/postproc/disallow_dynamic_loop.cc
  • Changed namespace from tvm::tir to tvm::s_tir.
  • Updated type references from tir::Schedule to s_tir::Schedule in Apply method.
  • Changed tir::DynamicExtentFinder::Find to s_tir::DynamicExtentFinder::Find.
• src/meta_schedule/postproc/postproc.cc
  • Updated type references from tir::Schedule to s_tir::Schedule in PyPostprocNode::Apply.
• src/meta_schedule/postproc/rewrite_cooperative_fetch.cc
  • Changed namespace from tvm::tir to tvm::s_tir.
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::Trace to s_tir::Trace, tir::Instruction to s_tir::Instruction, tir::SBlockRV to s_tir::SBlockRV, tir::LoopRV to s_tir::LoopRV.
  • Changed attr::meta_schedule_cooperative_fetch and attr::warp_execution to tir::attr::meta_schedule_cooperative_fetch and tir::attr::warp_execution.
  • Updated global function calls like tir::ParseThreadBinding, tir::ParseWarpExecutionAnn, tir::ParseAnnotate, tir::GetLoopIntExtent, tir::GetMaxUsedDtypeBytes to use s_tir namespace.
• src/meta_schedule/postproc/rewrite_layout.cc
  • Changed namespace from tvm::tir to tvm::s_tir.
  • Updated type references from tir::Schedule to s_tir::Schedule in Apply method.
  • Changed attr::layout_free_buffers and attr::meta_schedule_layout_rewrite_preproc to tir::attr::layout_free_buffers and tir::attr::meta_schedule_layout_rewrite_preproc.
  • Changed tir::RewriteLayout to s_tir::RewriteLayout.
• src/meta_schedule/postproc/rewrite_parallel_vectorize_unroll.cc
  • Changed namespace from tvm::tir to tvm::s_tir.
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::ParsedAnnotation to s_tir::ParsedAnnotation, tir::SBlockRV to s_tir::SBlockRV, tir::LoopRV to s_tir::LoopRV.
  • Changed attr::meta_schedule_parallel, attr::meta_schedule_vectorize, attr::meta_schedule_unroll_explicit, attr::meta_schedule_unroll_implicit, attr::pragma_auto_unroll_max_step, attr::pragma_unroll_explicit to their tir::attr counterparts.
  • Updated global function calls like tir::FindAnnotatedRootBlock, tir::AdjustParallelVectorize, tir::RewriteFuseSplitParallelVectorize, tir::RewriteParallel, tir::RewriteVectorize, tir::RewriteUnroll, tir::ScheduleError to use s_tir namespace.
• src/meta_schedule/postproc/rewrite_reduction_block.cc
  • Changed namespace from tvm::tir to tvm::s_tir.
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::SBlockRV to s_tir::SBlockRV, tir::LoopRV to s_tir::LoopRV.
  • Updated global function calls like tir::ReductionBlockFinder::Find, tir::FindDecomposePoint, tir::GetAnn to use s_tir namespace.
  • Changed tir::attr::meta_schedule_auto_tensorize and tir::attr::meta_schedule_auto_tensorize_init to their tir::attr counterparts.
• src/meta_schedule/postproc/rewrite_tensorize.cc
  • Updated type references from tir::LoopRV to s_tir::LoopRV, tir::SBlockRV to s_tir::SBlockRV, tir::Schedule to s_tir::Schedule.
  • Updated global function calls like tir::GetAnn, tir::TileWithTensorIntrin to use s_tir namespace.
  • Changed tir::attr::meta_schedule_auto_tensorize to tir::attr::meta_schedule_auto_tensorize.
• src/meta_schedule/postproc/rewrite_unbound_block.cc
  • Changed namespace from tvm::tir to tvm::s_tir.
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::ExprRV to s_tir::ExprRV, tir::LoopRV to s_tir::LoopRV, tir::SBlockRV to s_tir::SBlockRV.
  • Updated global function calls like tir::UnboundBlockFinder::Find to use s_tir namespace.
• src/meta_schedule/postproc/verify_gpu_code.cc
  • Changed namespace from tvm::tir to tvm::s_tir.
  • Updated type references from tir::Schedule to s_tir::Schedule.
  • Changed attr::warp_execution, attr::meta_schedule_thread_extent_low_inclusive, attr::meta_schedule_thread_extent_high_inclusive to their tir::attr counterparts.
  • Updated global function calls like tir::ThreadExtentChecker::Check to use s_tir namespace.
• src/meta_schedule/postproc/verify_vtcm_limit.cc
  • Updated type references from tir::Schedule to s_tir::Schedule in Apply method.
• src/meta_schedule/schedule/cpu/winograd.cc
  • Added using directives for s_tir::ExprRV, s_tir::LoopRV, s_tir::SBlockRV, s_tir::Schedule.
  • Updated type references from tir::LoopRV, tir::SBlockRV, tir::Schedule to their s_tir counterparts.
• src/meta_schedule/schedule/cuda/thread_bind.cc
  • Updated include path for schedule header to tvm/s_tir/schedule/schedule.h.
  • Added using directives for s_tir::ExprRV, s_tir::GetLoopIterType, s_tir::GetLoops, s_tir::GetThreadScope, s_tir::HasBeenMultiLevelTiled, s_tir::IsBlockIdx, s_tir::IsSingleStmt, s_tir::IsThreadIdx, s_tir::LoopRV, s_tir::SBlockRV, s_tir::Schedule.
  • Updated type references from tir::ExprRV, tir::LoopRV, tir::SBlockRV, tir::Schedule to their s_tir counterparts.
• src/meta_schedule/schedule/cuda/winograd.cc
  • Added using directives for s_tir::ExprRV, s_tir::LoopRV, s_tir::SBlockRV, s_tir::Schedule.
  • Updated type references from tir::LoopRV, tir::SBlockRV, tir::Schedule to their s_tir counterparts.
• src/meta_schedule/schedule/generic/winograd.cc
  • Added using directives for s_tir::ExprRV, s_tir::LoopRV, s_tir::SBlockRV, s_tir::Schedule.
• src/meta_schedule/schedule_rule/add_rfactor.cc
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::SBlockRV to s_tir::SBlockRV, tir::ExprRV to s_tir::ExprRV, tir::LoopRV to s_tir::LoopRV.
• src/meta_schedule/schedule_rule/apply_custom_rule.cc
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::SBlockRV to s_tir::SBlockRV.
  • Updated global function calls from tir::GetAnn to s_tir::GetAnn.
• src/meta_schedule/schedule_rule/auto_bind.cc
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::SBlockRV to s_tir::SBlockRV.
• src/meta_schedule/schedule_rule/auto_inline.cc
  • Added using directives for s_tir::AnalyzeReadWritePattern, s_tir::CanComputeInline, s_tir::CanReverseComputeInline, s_tir::GetAnn, s_tir::GetConsumers, s_tir::GetProducers, s_tir::GetRootPrimFunc, s_tir::GetSBlockRealize, s_tir::HasIfThenElse, s_tir::HasOp, s_tir::IsSpatialPrimFunc, s_tir::ScheduleState.
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::SBlockRV to s_tir::SBlockRV.
  • Updated global function calls like tir::GetAnn, tir::GetScopeRoot, tir::IsCompleteBlock to use s_tir namespace.
  • Changed tir::attr::meta_schedule_inline_rule and tir::attr::meta_schedule_auto_tensorize to their tir::attr counterparts.
• src/meta_schedule/schedule_rule/cross_thread_reduction.cc
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::LoopRV to s_tir::LoopRV, tir::SBlockRV to s_tir::SBlockRV, tir::ExprRV to s_tir::ExprRV, tir::Trace to s_tir::Trace.
  • Updated global function calls like tir::GetThreadScope, tir::IsThreadIdx, tir::GetLoopIntExtent, tir::SBlockRVs2StmtSRefs, tir::GetSRefLowestCommonAncestor, tir::GetLoopIterType to use s_tir namespace.
• src/meta_schedule/schedule_rule/multi_level_tiling.cc
  • Changed namespace from tvm::tir to tvm::s_tir.
  • Added using directives for s_tir::GetSBlockVarTypes, s_tir::IsWriteCache, s_tir::LoopRV, s_tir::SBlockRV, s_tir::Schedule.
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::SBlockRV to s_tir::SBlockRV, tir::LoopRV to s_tir::LoopRV, tir::ExprRV to s_tir::ExprRV.
  • Updated global function calls like tir::GetAnn, tir::GetLoopIntExtent to use s_tir namespace.
  • Changed tir::attr::meta_schedule_cooperative_fetch to tir::attr::meta_schedule_cooperative_fetch.
• src/meta_schedule/schedule_rule/multi_level_tiling.h
  • Updated include path for schedule header to tvm/s_tir/schedule/schedule.h.
  • Changed namespace from tvm::tir to tvm::s_tir.
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::SBlockRV to s_tir::SBlockRV, tir::LoopRV to s_tir::LoopRV, tir::ExprRV to s_tir::ExprRV.
• src/meta_schedule/schedule_rule/multi_level_tiling_tensor_core.cc
  • Added using directives for s_tir::GetSBlockVarTypes, s_tir::LoopRV, s_tir::SBlockRV, s_tir::Schedule.
  • Updated type references from tir::AutoTensorizeMappingInfo to s_tir::AutoTensorizeMappingInfo, tir::LoopRV to s_tir::LoopRV, tir::SBlockRV to s_tir::SBlockRV, tir::Schedule to s_tir::Schedule, tir::ExprRV to s_tir::ExprRV.
  • Updated global function calls like tir::GetAutoTensorizeMappingInfo, tir::TileWithTensorIntrin, tir::GetReadBufferNDims, tir::GetNthAccessBuffer, tir::GetNthAccessBufferRegion, tir::TileWithTensorIntrin, tir::GetLoopIntExtent to use s_tir namespace.
  • Changed tir::attr::meta_schedule_tiling_structure and tir::attr::vector_bytes to their tir::attr counterparts.
• src/meta_schedule/schedule_rule/multi_level_tiling_wide_vector.cc
  • Updated include paths for analysis.h and transform.h to s_tir namespace.
  • Updated using directives for LoopRV, SBlockRV, Schedule to s_tir namespace.
  • Updated type references from tir::ExprRV to s_tir::ExprRV.
  • Updated global function calls like tir::GetSBlockRealize, tir::GetLoopIntExtent to use s_tir namespace.
• src/meta_schedule/schedule_rule/multi_level_tiling_with_intrin.cc
  • Updated include paths for analysis.h and transform.h to s_tir namespace.
  • Updated type references from tir::SBlockRV to s_tir::SBlockRV, tir::LoopRV to s_tir::LoopRV, tir::Schedule to s_tir::Schedule.
  • Updated global function calls from tir::TileWithTensorIntrin to s_tir::TileWithTensorIntrin.
  • Changed tir::attr::meta_schedule_auto_tensorize to tir::attr::meta_schedule_auto_tensorize.
• src/meta_schedule/schedule_rule/parallel_vectorize_unroll.cc
  • Changed namespace from tvm::tir to tvm::s_tir.
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::SBlockRV to s_tir::SBlockRV.
  • Updated global function calls like tir::IsRootBlock, tir::CheckSpatialPrimFunc to use s_tir namespace.
  • Changed tir::attr::meta_schedule_parallel and tir::attr::meta_schedule_vectorize to their tir::attr counterparts.
• src/meta_schedule/schedule_rule/random_compute_location.cc
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::SBlockRV to s_tir::SBlockRV, tir::LoopRV to s_tir::LoopRV.
  • Updated global function calls like tir::HasAnn, tir::GetScopeRoot, tir::GetLoops, tir::GetChildBlockSRefOnSRefTree, tir::HasBeenMultiLevelTiled, tir::GetConsumers, tir::SampleComputeLocation to use s_tir namespace.
  • Changed tir::attr::meta_schedule_random_compute_producer to tir::attr::meta_schedule_random_compute_producer.
• src/meta_schedule/schedule_rule/schedule_rule.cc
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::SBlockRV to s_tir::SBlockRV.
• src/meta_schedule/search_strategy/evolutionary_search.cc
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::Trace to s_tir::Trace.
  • Updated global function calls like tir::MakeMultinomialSampler, tir::SampleInt, tir::SampleWithoutReplacement to use s_tir namespace.
• src/meta_schedule/search_strategy/replay_func.cc
  • Updated type references from tir::Schedule to s_tir::Schedule.
  • Updated global function calls from tir::SampleInt to s_tir::SampleInt.
• src/meta_schedule/search_strategy/replay_trace.cc
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::Trace to s_tir::Trace.
  • Updated global function calls from tir::SampleInt to s_tir::SampleInt.
• src/meta_schedule/search_strategy/search_strategy.cc
  • Updated type references from tir::Schedule to s_tir::Schedule.
• src/meta_schedule/space_generator/post_order_apply.cc
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::SBlockRV to s_tir::SBlockRV.
  • Updated global function calls from tir::GetAnn to s_tir::GetAnn.
• src/meta_schedule/space_generator/schedule_fn.cc
  • Updated type references from tir::Schedule to s_tir::Schedule.
• src/meta_schedule/space_generator/space_generator.cc
  • Updated type references from tir::Schedule to s_tir::Schedule.
• src/meta_schedule/space_generator/space_generator_union.cc
  • Updated type references from tir::Schedule to s_tir::Schedule.
• src/meta_schedule/task_scheduler/gradient_based.cc
  • Updated global function calls from tir::SampleInt to s_tir::SampleInt.
• src/meta_schedule/task_scheduler/task_scheduler.cc
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::Trace to s_tir::Trace.
• src/meta_schedule/trace_apply.cc
  • Updated include path for analysis header to s_tir namespace.
  • Added using directives for s_tir::GetSBlockNames, s_tir::Instruction, s_tir::InstructionKind, s_tir::IsSpatial, s_tir::LoopRV, s_tir::LoopRVNode, s_tir::SBlockRV, s_tir::SBlockRVNode, s_tir::Schedule, s_tir::Trace, s_tir::TranslateAddOutputRVs, s_tir::TranslateInputRVs.
• src/meta_schedule/trace_apply.h
  • Updated include paths for schedule and trace headers to tvm/s_tir/schedule/schedule.h and tvm/s_tir/schedule/trace.h.
  • Updated type references from tir::Schedule to s_tir::Schedule and tir::Trace to s_tir::Trace.
• src/meta_schedule/utils.h
  • Updated include paths for schedule and primitive headers to tvm/s_tir/schedule/schedule.h, s_tir/schedule/primitive.h, and s_tir/schedule/utils.h.
  • Updated inline function GetRVFromSRef to use s_tir::SBlockRV and s_tir::Schedule.
  • Updated ThreadedTraceApply struct to use s_tir::Schedule and s_tir::Trace.
  • Updated global function calls from tir::ScheduleError to s_tir::ScheduleError.
  • Updated SBlockCollector class to use s_tir::SBlockRV and s_tir::Schedule.
• src/relax/distributed/transform/legalize_redistribute.cc
  • Updated include path for schedule transform header to s_tir namespace.
• src/relax/distributed/transform/lower_distir.cc
  • Updated include path for schedule transform header to s_tir namespace.
• src/relax/distributed/transform/lower_global_view_to_local_view.cc
  • Updated include path for schedule transform header to s_tir namespace.
  • Added using directive for s_tir::ReplaceBuffer.
• src/relax/transform/meta_schedule.cc
  • Updated type references from tir::Schedule to s_tir::Schedule.
  • Updated global function calls from tir::Schedule::Traced to s_tir::Schedule::Traced.
  • Changed tir::ScheduleErrorRenderLevel::kDetail to s_tir::ScheduleErrorRenderLevel::kDetail.
• src/relax/transform/split_call_tir_by_pattern.cc
  • Updated include path for schedule ir_comparator header to s_tir namespace.
  • Added using directives for s_tir::ExprComparator, s_tir::TensorizeComparator.
• src/s_tir/transform/compact_buffer_region.cc
  • Updated include path for schedule utils header from ../../tir/schedule/utils.h to ../schedule/utils.h.
• src/s_tir/transform/inject_software_pipeline.cc
  • Updated include path for schedule utils header from ../../tir/schedule/utils.h to ../schedule/utils.h.
• src/s_tir/transform/lower_cross_thread_reduction.cc
  • Updated include path for schedule analysis header from ../../tir/schedule/analysis.h to ../schedule/analysis.h.
• src/s_tir/transform/manifest_shared_memory_local_stage.cc
  • Updated include path for schedule transform header from ../../tir/schedule/transform.h to ../schedule/transform.h.
• src/s_tir/transform/memhammer_lower_auto_copy.cc
  • Updated include path for schedule utils header from ../../tir/schedule/utils.h to ../schedule/utils.h.
• src/s_tir/transform/memhammer_rewrite_rule.h
  • Updated include path for schedule utils header from ../../tir/schedule/utils.h to ../schedule/utils.h.
• src/tir/analysis/oob_checker.cc
  • Updated include path for schedule error header from ../schedule/error.h to ../../s_tir/schedule/error.h.
  • Updated OOBError class to inherit from s_tir::ScheduleError.
• src/tir/schedule/analysis.h
  • File renamed to src/s_tir/schedule/analysis.h.
  • Updated include paths for schedule and state headers to tvm/s_tir/schedule/schedule.h and tvm/s_tir/schedule/state.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated TVM_FFI_DECLARE_OBJECT_INFO_FINAL macros to use s_tir.
  • Updated type references from tir::ScheduleState to s_tir::ScheduleState.
  • Updated global function calls from tir::GetLoops to GetLoops.
• src/tir/schedule/analysis/analysis.cc
  • File renamed to src/s_tir/schedule/analysis/analysis.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated TVM_FFI_STATIC_INIT_BLOCK definitions from tir.schedule.X to s_tir.schedule.X.
  • Updated global function calls from tir::GetAnn to GetAnn.
• src/tir/schedule/analysis/layout.cc
  • File renamed to src/s_tir/schedule/analysis/layout.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated TVM_FFI_STATIC_INIT_BLOCK definitions from tir.schedule.SuggestIndexMap to s_tir.schedule.SuggestIndexMap.
• src/tir/schedule/analysis/reducer.cc
  • File renamed to src/s_tir/schedule/analysis/reducer.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
• src/tir/schedule/analysis/verify.cc
  • File renamed to src/s_tir/schedule/analysis/verify.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
• src/tir/schedule/concrete_schedule.cc
  • File renamed to src/s_tir/schedule/concrete_schedule.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated global function calls from tir::SampleCategorical, tir::SamplePerfectTile, tir::SamplePartitionedTile, tir::SampleComputeLocation, tir::GetSBlocks, tir::GetLoops, tir::GetChildBlocks, tir::GetProducers, tir::GetConsumers, tir::GetOutputBlocks, tir::Merge, tir::Fuse, tir::Split, tir::LoopPartition, tir::Reorder, tir::ReorderBlockIterVar, tir::AddUnitLoop, tir::Parallel, tir::Vectorize, tir::Bind, tir::Unroll, tir::CacheRead, tir::CacheWrite, tir::ReindexCacheRead, tir::ReindexCacheWrite, tir::CacheInplace, tir::CacheIndex, tir::ReIndex, tir::ReadAt, tir::WriteAt, tir::ComputeInline, tir::ReverseComputeInline, tir::ComputeAt, tir::ReverseComputeAt, tir::FuseReductionEpilogue, tir::StorageAlign, tir::SetScope, tir::UnsafeSetDType, tir::DecomposeReduction, tir::RFactor, tir::Blockize, tir::Tensorize, tir::Annotate, tir::Unannotate, tir::TransformLayout, tir::TransformBlockLayout, tir::SetAxisSeparator, tir::DecomposePadding, tir::PadEinsum, tir::RollingBuffer, tir::UnsafeHideBufferAccess, tir::AnnotateBufferAccess to their s_tir counterparts.
  • Changed tir::ScheduleErrorRenderLevel::kDetail to s_tir::ScheduleErrorRenderLevel::kDetail.
• src/tir/schedule/concrete_schedule.h
  • File renamed to src/s_tir/schedule/concrete_schedule.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
• src/tir/schedule/error.cc
  • File renamed to src/s_tir/schedule/error.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
• src/tir/schedule/error.h
  • File renamed to src/s_tir/schedule/error.h.
  • Updated include path for state header to tvm/s_tir/schedule/state.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
• src/tir/schedule/instruction.cc
  • File renamed to src/s_tir/schedule/instruction.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated TVM_FFI_STATIC_INIT_BLOCK definitions from tir.schedule.X to s_tir.schedule.X.
• src/tir/schedule/instruction_traits.h
  • File renamed to src/s_tir/schedule/instruction_traits.h.
  • Updated include paths for instruction and schedule headers to tvm/s_tir/schedule/instruction.h and tvm/s_tir/schedule/schedule.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated type references from tir::Schedule to s_tir::Schedule.
• src/tir/schedule/ir_comparator.cc
  • File renamed to src/s_tir/schedule/ir_comparator.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
• src/tir/schedule/ir_comparator.h
  • File renamed to src/s_tir/schedule/ir_comparator.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
• src/tir/schedule/primitive.h
  • File renamed to src/s_tir/schedule/primitive.h.
  • Updated include path for state header to tvm/s_tir/schedule/state.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated type references from tir::ScheduleState to s_tir::ScheduleState.
• src/tir/schedule/primitive/annotate.cc
  • File renamed to src/s_tir/schedule/primitive/annotate.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/primitive/annotate_buffer_access.cc
  • File renamed to src/s_tir/schedule/primitive/annotate_buffer_access.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Changed attr::explicit_write_region and attr::explicit_read_region to tir::attr::explicit_write_region and tir::attr::explicit_read_region.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/primitive/block_annotate.cc
  • File renamed to src/s_tir/schedule/primitive/block_annotate.cc.
  • Updated include path for ir_utils header to ../../../tir/transforms/ir_utils.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Changed attr::buffer_dim_align to tir::attr::buffer_dim_align.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/primitive/blockize_tensorize.cc
  • File renamed to src/s_tir/schedule/primitive/blockize_tensorize.cc.
  • Updated include path for simplify header to ../../../tir/transforms/simplify.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated global function calls from UsesVar, tir::GetScopeRoot, tir::GetSBlockRealize to their s_tir counterparts.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/primitive/cache_index.cc
  • File renamed to src/s_tir/schedule/primitive/cache_index.cc.
  • Updated include paths for common_subexpr_elim_tools and replace_selected_expr headers to ../../../tir/transforms/common_subexpr_elim_tools.h and ../../../tir/transforms/replace_selected_expr.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/primitive/cache_read_write.cc
  • File renamed to src/s_tir/schedule/primitive/cache_read_write.cc.
  • Updated include paths for var_use_def_analysis and ir_utils headers to ../../../tir/analysis/var_use_def_analysis.h and ../../../tir/transforms/ir_utils.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated global function calls from tir::GetChildBlocks to GetChildBlocks.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/primitive/compute_at.cc
  • File renamed to src/s_tir/schedule/primitive/compute_at.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/primitive/compute_inline.cc
  • File renamed to src/s_tir/schedule/primitive/compute_inline.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/primitive/decompose_padding.cc
  • File renamed to src/s_tir/schedule/primitive/decompose_padding.cc.
  • Updated include path for ir_utils header to ../../../tir/transforms/ir_utils.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated TVM_FFI_STATIC_INIT_BLOCK definitions from tir.schedule.CanDecomposePadding to s_tir.schedule.CanDecomposePadding.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/primitive/for_kind.cc
  • File renamed to src/s_tir/schedule/primitive/for_kind.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/primitive/get_block_loop.cc
  • File renamed to src/s_tir/schedule/primitive/get_block_loop.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated global function calls from tir::GetProducers, tir::GetConsumers, tir::GetOutputBlocks to their s_tir counterparts.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/primitive/hide_buffer_access.cc
  • File renamed to src/s_tir/schedule/primitive/hide_buffer_access.cc.
  • Updated include path for ir_utils header to ../../../tir/transforms/ir_utils.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/primitive/layout_transformation.cc
  • File renamed to src/s_tir/schedule/primitive/layout_transformation.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/primitive/loop_transformation.cc
  • File renamed to src/s_tir/schedule/primitive/loop_transformation.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated global function calls from tir::GetScopeRoot, tir::GetSBlockRealize to their s_tir counterparts.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/primitive/pad_einsum.cc
  • File renamed to src/s_tir/schedule/primitive/pad_einsum.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/primitive/read_write_at.cc
  • File renamed to src/s_tir/schedule/primitive/read_write_at.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/primitive/reduction.cc
  • File renamed to src/s_tir/schedule/primitive/reduction.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated TVM_FFI_STATIC_INIT_BLOCK definitions from tir.schedule.RegisterReducer to s_tir.schedule.RegisterReducer.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/primitive/reorder_block_iter_var.cc
  • File renamed to src/s_tir/schedule/primitive/reorder_block_iter_var.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/primitive/rolling_buffer.cc
  • File renamed to src/s_tir/schedule/primitive/rolling_buffer.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated global function calls from tir::GetSBlockRealize to GetSBlockRealize.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/primitive/sampling.cc
  • File renamed to src/s_tir/schedule/primitive/sampling.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated global function calls from tir::SampleCategorical, tir::SamplePerfectTile, tir::SamplePartitionedTile, tir::SampleComputeLocation to their s_tir counterparts.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/schedule.cc
  • File renamed to src/s_tir/schedule/schedule.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated TVM_FFI_STATIC_INIT_BLOCK definitions from tir.schedule.X to s_tir.schedule.X.
• src/tir/schedule/state.cc
  • File renamed to src/s_tir/schedule/state.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated TVM_FFI_STATIC_INIT_BLOCK definitions from tir.schedule.X to s_tir.schedule.X.
• src/tir/schedule/trace.cc
  • File renamed to src/s_tir/schedule/trace.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated TVM_FFI_STATIC_INIT_BLOCK definitions from tir.schedule.X to s_tir.schedule.X.
  • Updated friend struct ::tvm::tir::UnpackedInstTraits to friend struct ::tvm::s_tir::UnpackedInstTraits.
• src/tir/schedule/traced_schedule.cc
  • File renamed to src/s_tir/schedule/traced_schedule.cc.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated global function calls from tir::SampleCategorical, tir::SamplePerfectTile, tir::SamplePartitionedTile, tir::SampleComputeLocation to their s_tir counterparts.
• src/tir/schedule/traced_schedule.h
  • File renamed to src/s_tir/schedule/traced_schedule.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
• src/tir/schedule/transform.cc
  • File renamed to src/s_tir/schedule/transform.cc.
  • Updated include path for ir_utils header to ../../../tir/transforms/ir_utils.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated type references from tir::TensorizeInfo, tir::TensorizeInfoNode to s_tir::TensorizeInfo, s_tir::TensorizeInfoNode.
  • Updated global function calls from tir::GetTensorizeLoopMapping to GetTensorizeLoopMapping.
  • Updated TVM_FFI_STATIC_INIT_BLOCK definitions from tir.schedule.TileWithTensorIntrin to s_tir.schedule.TileWithTensorIntrin.
  • Updated TVM_FFI_STATIC_INIT_BLOCK definitions from tir.schedule.NormalizePrimFunc to s_tir.schedule.NormalizePrimFunc.
• src/tir/schedule/transform.h
  • File renamed to src/s_tir/schedule/transform.h.
  • Updated include paths for schedule and state headers to tvm/s_tir/schedule/schedule.h and tvm/s_tir/schedule/state.h.
  • Updated include path for functor_common header to ../../tir/ir/functor_common.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated type references from tir::LoopRV to s_tir::LoopRV, tir::SBlockRV to s_tir::SBlockRV, tir::Schedule to s_tir::Schedule.
• src/tir/schedule/utils.h
  • File renamed to src/s_tir/schedule/utils.h.
  • Updated include paths for instruction, schedule, state, and trace headers to tvm/s_tir/schedule/instruction.h, tvm/s_tir/schedule/schedule.h, tvm/s_tir/schedule/state.h, and tvm/s_tir/schedule/trace.h.
  • Namespace changed from tvm::tir to tvm::s_tir.
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::SBlockRV to s_tir::SBlockRV, tir::LoopRV to s_tir::LoopRV.
  • Updated global function calls from tir::GetLoopIterType to GetLoopIterType.
• src/tir/transforms/default_gpu_schedule.cc
  • Updated type references from tir::Schedule to s_tir::Schedule, tir::SBlockRV to s_tir::SBlockRV, tir::LoopRV to s_tir::LoopRV.
  • Changed tir::ScheduleErrorRenderLevel::kDetail to s_tir::ScheduleErrorRenderLevel::kDetail.
• tests/python/dlight/test_primitives.py
  • Updated global function call from tvm.get_global_func("tir.schedule.NormalizePrimFunc") to tvm.get_global_func("s_tir.schedule.NormalizePrimFunc").

Activity

• The pull request was created by tqchen.
• The pull request description indicates that it lifts the previous tir/schedule to s_tir namespace.
• No other human activity (comments, reviews, etc.) has been recorded for this pull request.

Using Gemini Code Assist

The full guide for Gemini Code Assist can be found on our documentation page, here are some quick tips.

Invoking Gemini

You can request assistance from Gemini at any point by creating a comment using either /gemini <command> or @gemini-code-assist <command>. Below is a summary of the supported commands on the current page.

Feature	Command	Description
Code Review	/gemini review	Performs a code review for the current pull request in its current state.
Pull Request Summary	/gemini summary	Provides a summary of the current pull request in its current state.
Comment	@gemini-code-assist	Responds in comments when explicitly tagged, both in pull request comments and review comments.
Help	/gemini help	Displays a list of available commands.

Customization

To customize Gemini Code Assist for GitHub experience, repository maintainers can create a configuration file and/or provide a custom code review style guide (such as PEP-8 for Python) by creating and adding files to a .gemini/ folder in the base of the repository. Detailed instructions can be found here.

Limitations & Feedback

Gemini Code Assist may make mistakes. Please leave feedback on any instances where its feedback is incorrect or counter productive. You can react with 👍 and 👎 on @gemini-code-assist comments. If you're interested in giving your feedback about your experience with Gemini Code Assist for Github and other Google products, sign up here.

You can also get AI-powered code generation, chat, as well as code reviews directly in the IDE at no cost with the Gemini Code Assist IDE Extension.

Footnotes

1. Review the Privacy Notices, Generative AI Prohibited Use Policy, Terms of Service, and learn how to configure Gemini Code Assist in GitHub here. Gemini can make mistakes, so double check it and use code with caution. ↩

[gemini-code-assist]

Code Review

This pull request is a large-scale refactoring that migrates the TensorIR scheduling infrastructure from the tir::schedule namespace to a new s_tir namespace. The changes are extensive and touch many files across the C++ and Python codebase. This includes renaming header and source files, updating namespaces, and adjusting FFI registrations and object names. The refactoring is done systematically and consistently. I've reviewed the changes and they appear correct, contributing to better code organization. I did not find any issues or bugs introduced by this refactoring.