This is an automated email from the ASF dual-hosted git repository.
tlopex 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 ec7f59f2d4 [TVMScript] Add test for TIR macro block name suffix
handling (#18504)
ec7f59f2d4 is described below
commit ec7f59f2d4dcb649573cd7de6551e588702f8164
Author: Guan-Ming (Wesley) Chiu <[email protected]>
AuthorDate: Wed Nov 26 13:16:18 2025 +0800
[TVMScript] Add test for TIR macro block name suffix handling (#18504)
## How
add missing tests for https://github.com/apache/tvm/pull/18465
---
.../python/tvmscript/test_tvmscript_parser_tir.py | 32 ++++++++++++++++++++++
1 file changed, 32 insertions(+)
diff --git a/tests/python/tvmscript/test_tvmscript_parser_tir.py
b/tests/python/tvmscript/test_tvmscript_parser_tir.py
index 3b84e919c8..cc285e9835 100644
--- a/tests/python/tvmscript/test_tvmscript_parser_tir.py
+++ b/tests/python/tvmscript/test_tvmscript_parser_tir.py
@@ -638,6 +638,38 @@ def test_alloc_inside_block():
tvm.ir.assert_structural_equal(func, expected)
+def test_tir_macro_block_name_suffix():
+ @T.macro
+ def operation(A, idx):
+ with T.block("op"):
+ v = T.axis.remap("S", [idx])
+ A[v] = A[v] * T.float32(2)
+
+ @T.prim_func(private=True)
+ def func_w_macro(a: T.handle) -> None:
+ A = T.match_buffer(a, [10])
+ for i in T.serial(0, 10):
+ operation(A, i)
+ operation(A, i)
+ operation(A, i)
+
+ @T.prim_func(private=True)
+ def expected(a: T.handle) -> None:
+ A = T.match_buffer(a, [10])
+ for i in T.serial(0, 10):
+ with T.block("op"):
+ v = T.axis.remap("S", [i])
+ A[v] = A[v] * T.float32(2)
+ with T.block("op_1"):
+ v = T.axis.remap("S", [i])
+ A[v] = A[v] * T.float32(2)
+ with T.block("op_2"):
+ v = T.axis.remap("S", [i])
+ A[v] = A[v] * T.float32(2)
+
+ tvm.ir.assert_structural_equal(func_w_macro, expected)
+
+
def test_ifexp():
@T.prim_func(private=True)
def func(A: T.buffer((128, 128), "float32")):
