Skip to content

Commit 947aafc

Browse files
authored
1 parent 829b8cb commit 947aafc

File tree

3 files changed

+10
-2
lines changed

3 files changed

+10
-2
lines changed

src/kirin/dialects/ilist/rewrite/inline_getitem.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,9 @@ def rewrite_Statement(self, node: ir.Statement) -> abc.RewriteResult:
2525
if not isinstance(index_const := node.index.hints.get("const"), const.Value):
2626
return abc.RewriteResult()
2727

28+
if not node.result.uses:
29+
return abc.RewriteResult()
30+
2831
index = index_const.data
2932
if isinstance(index, int) and (
3033
0 <= index < len(stmt.args) or -len(stmt.args) <= index < 0

src/kirin/rewrite/getitem.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,9 @@ def rewrite_Statement(self, node: ir.Statement) -> RewriteResult:
1919
if not isinstance(index_value := node.index.hints.get("const"), const.Value):
2020
return RewriteResult()
2121

22+
if not node.result.uses:
23+
return RewriteResult()
24+
2225
stmt = node.obj.owner
2326
index = index_value.data
2427
if isinstance(index, int) and (

test/dialects/test_ilist.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -207,7 +207,8 @@ def test_inline_getitem_slice():
207207
[
208208
qreg := ilist.New(values=values),
209209
slice_value := py.Constant(slice(2, 5, 1)),
210-
py.GetItem(obj=qreg.result, index=slice_value.result),
210+
res := py.GetItem(obj=qreg.result, index=slice_value.result),
211+
func.Return(res.result),
211212
]
212213
)
213214
slice_value.result.hints["const"] = const.Value(slice(2, 5, 1))
@@ -219,7 +220,8 @@ def test_inline_getitem_slice():
219220
[
220221
qreg := ilist.New(values=values),
221222
slice_value := py.Constant(slice(2, 5, 1)),
222-
ilist.New(values=(values[2], values[3], values[4])),
223+
res := ilist.New(values=(values[2], values[3], values[4])),
224+
func.Return(res.result),
223225
]
224226
)
225227
assert test_block.is_equal(expected_block)

0 commit comments

Comments
 (0)