Skip to content

Commit

Permalink
[Relax][BugFix] Fix a bug about the IR construction in test file (#17121
Browse files Browse the repository at this point in the history
)

Update test_transform_dead_code_elimination.py

Fix the wrong Relax IR construction
  • Loading branch information
Cookiee235 authored Jul 18, 2024
1 parent 73078f1 commit 22a8978
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/python/relax/test_transform_dead_code_elimination.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand All @@ -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(
Expand Down

0 comments on commit 22a8978

Please sign in to comment.