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 <[email protected]>
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)

Reply via email to