You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by tq...@apache.org on 2026/09/23 09:15:06 UTC
(tvm) 05/11: Make conditional dataflow results explicit in fusion coverage
This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch script/canonical-parser-df
in repository https://gitbox.apache.org/repos/asf/tvm.git
commit 0ff2d0a3e9e5dd3f99484dd70628c3ccde7d4fe3
Author: Tianqi Chen <tq...@users.noreply.github.com>
AuthorDate: Wed Sep 23 08:43:19 2026 +0000
Make conditional dataflow results explicit in fusion coverage
---
tests/python/relax/test_transform_fuse_ops_by_pattern.py | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/tests/python/relax/test_transform_fuse_ops_by_pattern.py b/tests/python/relax/test_transform_fuse_ops_by_pattern.py
index 3e889f9234..f3ca25ad9d 100644
--- a/tests/python/relax/test_transform_fuse_ops_by_pattern.py
+++ b/tests/python/relax/test_transform_fuse_ops_by_pattern.py
@@ -1269,12 +1269,14 @@ def test_dataflow_inside_branch():
if transpose_weights:
with R.dataflow():
w_t = R.permute_dims(w)
- out = R.matmul(x, w_t)
- R.output(out)
+ out_then = R.matmul(x, w_t)
+ R.output(out_then)
+ out = out_then
else:
with R.dataflow():
- out = R.matmul(x, w)
- R.output(out)
+ out_else = R.matmul(x, w)
+ R.output(out_else)
+ out = out_else
return out
@I.ir_module(s_tir=True)