From 247b1c404d7afc10505157a434370e73e1ad5548 Mon Sep 17 00:00:00 2001
From: zackcquic <82788137+zackcquic@users.noreply.github.com>
Date: Mon, 21 Jun 2021 09:56:01 +0800
Subject: [PATCH] [Relay][Dataflow] Fix test_rewrite_function_with_fuzzy_body
 test check (#8287)

---
 tests/python/relay/test_dataflow_pattern.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/python/relay/test_dataflow_pattern.py b/tests/python/relay/test_dataflow_pattern.py
index f95a009f9dff..1c721f40d129 100644
--- a/tests/python/relay/test_dataflow_pattern.py
+++ b/tests/python/relay/test_dataflow_pattern.py
@@ -1609,7 +1609,7 @@ def callback(self, pre, post, node_map):
             return x + w
 
     out = rewrite(TestRewrite(), expr)
-    assert tvm.ir.structural_equal(x + w, x + w)
+    assert tvm.ir.structural_equal(out, x + w + b)
 
 
 def test_partition_function_with_fuzzy_body():