(tvm) branch main updated: [Relax][BugFix] Fix a bug about the IR construction in test file (#17121)

[syfeng]

This is an automated email from the ASF dual-hosted git repository.

syfeng pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 22a89785ba [Relax][BugFix] Fix a bug about the IR construction in test 
file  (#17121)
22a89785ba is described below

commit 22a89785bab2e120bb089a2d617342db0d157bc7
Author: Cookiee235 <qingc...@tju.edu.cn>
AuthorDate: Thu Jul 18 21:49:11 2024 +0800

    [Relax][BugFix] Fix a bug about the IR construction in test file  (#17121)
    
    Update test_transform_dead_code_elimination.py
    
    Fix the wrong Relax IR construction
---
 tests/python/relax/test_transform_dead_code_elimination.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/python/relax/test_transform_dead_code_elimination.py 
b/tests/python/relax/test_transform_dead_code_elimination.py
index 0cb0d46247..142faf5160 100644
--- 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 test_unused_dfb2():
                 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 test_unused_dfb2():
             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(