Fix unit test

apache · Oct 23, 2023 · 5ada41f · 5ada41f
1 parent b3f0a94
commit 5ada41f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/python/relax/test_dataflow_pattern.py b/tests/python/relax/test_dataflow_pattern.py
@@ -1404,9 +1404,8 @@ def before(x: R.Tensor((1024,))):
         def expected(x: R.Tensor((1024,))):
             with R.dataflow():
                 a = R.add(x, x)
-                b = a
-                R.output(b)
-            return b
+                R.output(a)
+            return a
 
     else: