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 564cd8808b [Relax] Update dropout call_tir out_ty spelling (#19874)
564cd8808b is described below
commit 564cd8808bf4300ac13f5f0f5ab2dcd26629bb14
Author: Tianqi Chen <[email protected]>
AuthorDate: Tue Jun 23 12:23:58 2026 -0400
[Relax] Update dropout call_tir out_ty spelling (#19874)
This updates the remaining dropout expected TVMScript after the Relax
call_tir spelling changed from out_sinfo to out_ty.
The affected call returns two tensors, matching nearby multi-output
call_tir sites that already use out_ty=[...].
---
tests/python/relax/test_transform_legalize_ops_nn.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/python/relax/test_transform_legalize_ops_nn.py
b/tests/python/relax/test_transform_legalize_ops_nn.py
index 88621b9067..284a0adcb9 100644
--- a/tests/python/relax/test_transform_legalize_ops_nn.py
+++ b/tests/python/relax/test_transform_legalize_ops_nn.py
@@ -4198,7 +4198,7 @@ def test_dropout():
@R.function
def main(x: R.Tensor((2, 3), dtype="float32")) -> R.Tuple(R.Tensor((2,
3), dtype="float32"), R.Tensor((2, 3), dtype="float32")):
cls = Expected
- gv = R.call_tir(cls.dropout, (x,), out_sinfo=[R.Tensor((2, 3),
dtype="float32"), R.Tensor((2, 3), dtype="float32")])
+ gv = R.call_tir(cls.dropout, (x,), out_ty=[R.Tensor((2, 3),
dtype="float32"), R.Tensor((2, 3), dtype="float32")])
return gv
# fmt: on
