From 18642a866e70e76d155339dc754db063295586e2 Mon Sep 17 00:00:00 2001 From: tlopex <820958424@qq.com> Date: Sat, 30 May 2026 00:17:54 -0400 Subject: [PATCH] [TIRX] Fix stale Simplify import in lowering test test_transform_lower_tirx.py imports and calls Simplify, but the pass is named StmtSimplify (the only simplify pass exported from tvm.tirx.transform). The stale name makes the module fail to import at collection time. Use StmtSimplify so the test collects and runs. --- tests/python/tirx/transform/test_transform_lower_tirx.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/python/tirx/transform/test_transform_lower_tirx.py b/tests/python/tirx/transform/test_transform_lower_tirx.py index 3e20d61f8059..80e68243d0b3 100644 --- a/tests/python/tirx/transform/test_transform_lower_tirx.py +++ b/tests/python/tirx/transform/test_transform_lower_tirx.py @@ -24,7 +24,7 @@ from tvm.tirx.layout import laneid, warpid, wg_local_layout from tvm.tirx.stmt import ExecScopeStmt from tvm.tirx.stmt_functor import post_order_visit -from tvm.tirx.transform import LowerTIRx, Simplify +from tvm.tirx.transform import LowerTIRx, StmtSimplify def _contains_exec_scope(mod): @@ -1000,7 +1000,7 @@ def before(A_ptr: Tx.handle): with tvm.target.Target("cuda"): lowered = LowerTIRx()(tvm.IRModule({"main": before})) - simplified = Simplify()(lowered) + simplified = StmtSimplify()(lowered) script = simplified.script(extra_config={"tirx.prefix": "Tx"}) assert "if warp_id_in_cta // 4 == 0:" in script