This is an automated email from the ASF dual-hosted git repository. tqchen pushed a commit to branch tvm-further-cleanup-python-tests-followup in repository https://gitbox.apache.org/repos/asf/tvm.git
commit b7642c49339f487b329127873b15c6e13d324f97 Author: Tianqi Chen <[email protected]> AuthorDate: Sun Jul 5 18:48:11 2026 +0000 [TEST][S-TIR] Type DLTensor shape pointers --- ...test_s_tir_transform_plan_update_buffer_allocation_location.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/python/s_tir/transform/test_s_tir_transform_plan_update_buffer_allocation_location.py b/tests/python/s_tir/transform/test_s_tir_transform_plan_update_buffer_allocation_location.py index d5173bcc13..7aabb5b897 100644 --- a/tests/python/s_tir/transform/test_s_tir_transform_plan_update_buffer_allocation_location.py +++ b/tests/python/s_tir/transform/test_s_tir_transform_plan_update_buffer_allocation_location.py @@ -384,9 +384,13 @@ def test_dltensor_buffer_is_unlowered(): @T.prim_func(s_tir=True) def before(dlpack_handle: T.handle, axis: T.int64) -> T.int64: ndim: T.int32 = T.tvm_struct_get(dlpack_handle, 0, 5, "int32") - stride_ptr: T.let[T.handle("int64")] = T.tvm_struct_get(dlpack_handle, 0, 4, "handle") + stride_ptr: T.let[T.handle("int64")] = T.tvm_struct_get( + dlpack_handle, 0, 4, tvm.ir.PointerType(tvm.ir.PrimType("int64")) + ) if T.isnullptr(stride_ptr): - shape_ptr: T.let[T.handle("int64")] = T.tvm_struct_get(dlpack_handle, 0, 3, "handle") + shape_ptr: T.let[T.handle("int64")] = T.tvm_struct_get( + dlpack_handle, 0, 3, tvm.ir.PointerType(tvm.ir.PrimType("int64")) + ) shape = T.decl_buffer(ndim, "int64", data=shape_ptr) product = T.decl_buffer([], "int64") product[()] = 1
