This is an automated email from the ASF dual-hosted git repository.
guan404ming 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 7a9f568049 [TIRx] Fix stale Simplify import in lowering test (#19642)
7a9f568049 is described below
commit 7a9f568049c2b44d93b1bf747df8113c693b5340
Author: Shushi Hong <[email protected]>
AuthorDate: Sat May 30 04:11:09 2026 -0400
[TIRx] Fix stale Simplify import in lowering test (#19642)
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 3e20d61f80..80e68243d0 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.function import PrimFunc
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
test_simplify_uses_floor_div_scope_predicate_as_context_fact():
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
