[Relay][Dataflow] Fix test_rewrite_function_with_fuzzy_body test check (

#8287)
apache · Jun 21, 2021 · 247b1c4 · 247b1c4
1 parent ca99552
commit 247b1c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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():