[Relay][Dataflow] Fix test_rewrite_function_with_fuzzy_body test check (

apache#8287)
zxy844288792 · Jan 13, 2022 · 74e7a3a · 74e7a3a
1 parent 2acaecf
commit 74e7a3a
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():