Update test_transform_dead_code_elimination.py

Fix the wrong Relax IR construction
apache · Jun 27, 2024 · 8b51526 · 8b51526
1 parent 63f9cd6
commit 8b51526
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/relax/test_transform_dead_code_elimination.py b/tests/python/relax/test_transform_dead_code_elimination.py
@@ -454,7 +454,7 @@ def main(
                 R.output(lv0)
 
             gv_x = R.astype(x, dtype="float16")
-            gv_w = R.astype(x, dtype="float16")
+            gv_w = R.astype(w, dtype="float16")
 
             with R.dataflow():
                 lv1: R.Tensor((2, 28, 28, 3), dtype="float16") = R.permute_dims(
@@ -481,7 +481,7 @@ def main(
             w: R.Tensor((4, 3, 3, 3), dtype="float32"),
         ):
             gv_x = R.astype(x, dtype="float16")
-            gv_w = R.astype(x, dtype="float16")
+            gv_w = R.astype(w, dtype="float16")
 
             with R.dataflow():
                 lv1: R.Tensor((2, 28, 28, 3), dtype="float16") = R.permute_dims(