Skip to content

Commit 564cd88

Browse files
authored
[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=[...].
1 parent 9808108 commit 564cd88

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tests/python/relax/test_transform_legalize_ops_nn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4198,7 +4198,7 @@ def dropout(x: T.Buffer((T.int64(2), T.int64(3)), "float32"), compute: T.Buffer(
41984198
@R.function
41994199
def main(x: R.Tensor((2, 3), dtype="float32")) -> R.Tuple(R.Tensor((2, 3), dtype="float32"), R.Tensor((2, 3), dtype="float32")):
42004200
cls = Expected
4201-
gv = R.call_tir(cls.dropout, (x,), out_sinfo=[R.Tensor((2, 3), dtype="float32"), R.Tensor((2, 3), dtype="float32")])
4201+
gv = R.call_tir(cls.dropout, (x,), out_ty=[R.Tensor((2, 3), dtype="float32"), R.Tensor((2, 3), dtype="float32")])
42024202
return gv
42034203
# fmt: on
42044204

0 commit comments

Comments
 (0)