This is an automated email from the ASF dual-hosted git repository.
tqchen 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 2eb788be11 [Fix][Tests] Update canonicalize bindings tests to T.Tensor
(#20541)
2eb788be11 is described below
commit 2eb788be110d83fbb04da848161e7feb62be8574
Author: Shushi Hong <[email protected]>
AuthorDate: Sun Oct 4 07:28:57 2026 -0400
[Fix][Tests] Update canonicalize bindings tests to T.Tensor (#20541)
The regression tests added in #20494 still use T.Buffer, which was
renamed to T.Tensor in #20519. Update their annotations so both tests
can construct their IR and exercise the canonicalizer.
---
tests/python/relax/test_transform_canonicalize_bindings.py | 12 ++++++------
1 file changed, 6 insertions(+), 6 deletions(-)
diff --git a/tests/python/relax/test_transform_canonicalize_bindings.py
b/tests/python/relax/test_transform_canonicalize_bindings.py
index 1a4c1d1b53..325d6d711b 100644
--- a/tests/python/relax/test_transform_canonicalize_bindings.py
+++ b/tests/python/relax/test_transform_canonicalize_bindings.py
@@ -1417,9 +1417,9 @@ def
test_call_tir_out_ty_follows_arguments_when_match_cast_is_kept():
@Ts.prim_func(private=True)
def transpose(
- x: T.Buffer((1, n, 2, m, 2, 8), "float16"),
- y: T.Buffer((1, n, m, 2, 2, 8), "float16"),
- z: T.Buffer((k,), "float16"),
+ x: T.Tensor((1, n, 2, m, 2, 8), "float16"),
+ y: T.Tensor((1, n, m, 2, 2, 8), "float16"),
+ z: T.Tensor((k,), "float16"),
):
for i0, i1, i2, i3, i4, i5 in T.grid(1, n, m, 2, 2, 8):
with Ts.sblock("b"):
@@ -1427,7 +1427,7 @@ def
test_call_tir_out_ty_follows_arguments_when_match_cast_is_kept():
y[v0, v1, v2, v3, v4, v5] = x[v0, v1, v3, v2, v4, v5]
@Ts.prim_func(private=True)
- def add_scalar(x: T.Buffer((1, 8), "float16"), s: T.int64, y:
T.Buffer((1, 8), "float16")):
+ def add_scalar(x: T.Tensor((1, 8), "float16"), s: T.int64, y:
T.Tensor((1, 8), "float16")):
for i in T.serial(8):
with Ts.sblock("b"):
vi = Ts.axis.spatial(8, i)
@@ -1475,14 +1475,14 @@ def
test_call_tir_out_ty_follows_arguments_through_chained_match_casts():
@I.ir_module
class Before:
@Ts.prim_func(private=True)
- def copy(x: T.Buffer((n, 8), "float16"), y: T.Buffer((n, 8),
"float16")):
+ def copy(x: T.Tensor((n, 8), "float16"), y: T.Tensor((n, 8),
"float16")):
for i, j in T.grid(n, 8):
with Ts.sblock("b"):
vi, vj = Ts.axis.remap("SS", [i, j])
y[vi, vj] = x[vi, vj]
@Ts.prim_func(private=True)
- def add_scalar(x: T.Buffer((1, 8), "float16"), s: T.int64, y:
T.Buffer((1, 8), "float16")):
+ def add_scalar(x: T.Tensor((1, 8), "float16"), s: T.int64, y:
T.Tensor((1, 8), "float16")):
for i in T.serial(8):
with Ts.sblock("b"):
vi = Ts.axis.spatial(8, i)
