This is an automated email from the ASF dual-hosted git repository. tqchen pushed a commit to branch script/canonical-parser-df in repository https://gitbox.apache.org/repos/asf/tvm.git
commit 2214eaf4d0738b5fdd2321486f726f1ae4420a3a Author: Tianqi Chen <[email protected]> AuthorDate: Wed Sep 23 08:48:05 2026 +0000 Keep local buffer names distinct from script namespaces --- tests/python/s_tir/schedule/test_tir_schedule_pad_einsum.py | 8 ++++---- tests/python/tirx/test_op_namespace_cleanup.py | 4 ++-- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/python/s_tir/schedule/test_tir_schedule_pad_einsum.py b/tests/python/s_tir/schedule/test_tir_schedule_pad_einsum.py index 74ba061367..a3fbbffb88 100644 --- a/tests/python/s_tir/schedule/test_tir_schedule_pad_einsum.py +++ b/tests/python/s_tir/schedule/test_tir_schedule_pad_einsum.py @@ -240,7 +240,7 @@ def test_pad_rms(): n = T.int32() A = T.match_buffer(a, (1, n, 4096)) W = T.match_buffer(w, (4096,), "float32") - R = T.match_buffer(r, (1, n, 4096), "float32") + Result = T.match_buffer(r, (1, n, 4096), "float32") S = T.sblock_alloc_buffer((1, n), "float32") for bsz, i, k in T.grid(1, n, 4096): with T.sblock("S"): @@ -253,7 +253,7 @@ def test_pad_rms(): for bsz, i, k in T.grid(1, n, 4096): with T.sblock("R"): v_bsz, v_i, v_k = T.axis.remap("SSS", [bsz, i, k]) - R[v_bsz, v_i, v_k] = W[v_k] * ( + Result[v_bsz, v_i, v_k] = W[v_k] * ( A[v_bsz, v_i, v_k] / T.sqrt(S[v_bsz, v_i] * T.float32(0.000244140625) + T.float32(1e-6)) ) @@ -264,7 +264,7 @@ def test_pad_rms(): n = T.int32() A = T.match_buffer(a, (1, n, 4096)) W = T.match_buffer(w, (4096,), "float32") - R = T.match_buffer(r, (1, n, 4096)) + Result = T.match_buffer(r, (1, n, 4096)) S = T.sblock_alloc_buffer((1, n)) A_pad = T.sblock_alloc_buffer((1, (n + 31) // 32 * 32, 4096)) S_pad = T.sblock_alloc_buffer((1, (n + 31) // 32 * 32)) @@ -289,7 +289,7 @@ def test_pad_rms(): for bsz, i, k in T.grid(1, n, 4096): with T.sblock("R"): v_bsz, v_i, v_k = T.axis.remap("SSS", [bsz, i, k]) - R[v_bsz, v_i, v_k] = W[v_k] * ( + Result[v_bsz, v_i, v_k] = W[v_k] * ( A[v_bsz, v_i, v_k] / T.sqrt(S[v_bsz, v_i] * T.float32(0.000244140625) + T.float32(1e-6)) ) diff --git a/tests/python/tirx/test_op_namespace_cleanup.py b/tests/python/tirx/test_op_namespace_cleanup.py index 15fd01448c..2f2d4ff862 100644 --- a/tests/python/tirx/test_op_namespace_cleanup.py +++ b/tests/python/tirx/test_op_namespace_cleanup.py @@ -293,9 +293,9 @@ def test_device_intrinsic_printer_roundtrips_canonical_namespaces(): @T.prim_func def device_namespaces(dst: T.handle, src: T.handle): A = T.match_buffer(src, (1,), "float32") - R = T.alloc_buffer((1,), "float32", scope="local") + Result = T.alloc_buffer((1,), "float32", scope="local") T.cuda.cta_sync() - T.s_tir.ldg32(R[0], 1, A[0], 0) + T.s_tir.ldg32(Result[0], 1, A[0], 0) T.metal.simd_shuffle(A[0], 0) T.metal.simd_shuffle_up(A[0], 1) T.metal.simd_shuffle_down(A[0], 1)
