From 2bdedc93aa6e130be17ad9e70cb17614863b132a Mon Sep 17 00:00:00 2001 From: tqchen Date: Tue, 23 Jun 2026 13:41:36 +0000 Subject: [PATCH] [FIX][Relax] Update dropout call_tir out_ty spelling --- 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 88621b9067df..284a0adcb9d8 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 dropout(x: T.Buffer((T.int64(2), T.int64(3)), "float32"), compute: T.Buffer( @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