Fixing bug in previous PR. (#481)

fixes issue in #479

Author: weinbe58

src/kirin/rewrite/cse.py

@@ -1,12 +1,15 @@
 from __future__ import annotations
 
-import sys
 from dataclasses import field, dataclass
 
 from kirin import ir
 from kirin.rewrite.abc import RewriteRule, RewriteResult
 
-_IS_PYTHON_310 = sys.version_info[0] == 3 and sys.version_info[1] == 10
+try:
+    hash(slice(None))
+    _HASHABLE_SLICE = True
+except TypeError:
+    _HASHABLE_SLICE = False
 
 
 @dataclass
@@ -22,7 +25,10 @@ class Info:
     _hashable: bool = field(init=False, repr=False)
 
     def __post_init__(self):
-        if _IS_PYTHON_310 and any(isinstance(attr, slice) for attr in self.attributes):
+        if not _HASHABLE_SLICE and any(
+            isinstance(attr, ir.PyAttr) and isinstance(attr.data, slice)
+            for attr in self.attributes
+        ):
             self._hash = id(self)
             self._hashable = False
         else:

test/rewrite/test_cse_rewrite.py

@@ -1,7 +1,7 @@
 from kirin import ir, lowering
 from kirin.decl import info, statement
 from kirin.prelude import basic_no_opt
-from kirin.rewrite.cse import CommonSubexpressionElimination
+from kirin.rewrite.cse import _HASHABLE_SLICE, Info, CommonSubexpressionElimination
 from kirin.rewrite.walk import Walk
 
 dialect = ir.Dialect("test")
@@ -35,3 +35,14 @@ def duplicated():
     stmt_1 = duplicated.callable_region.blocks[0].stmts.at(1)
     assert isinstance(stmt_0, MultiResult)
     assert not isinstance(stmt_1, MultiResult)
+
+
+def test_info():
+    info_value = Info(ir.Statement, (), (ir.PyAttr(slice(None)),), (), ())
+
+    if not _HASHABLE_SLICE:
+        assert info_value._hashable is False
+        assert info_value._hash == id(info_value)
+    else:
+        assert info_value._hashable is True
+        info_value._hash = hash((ir.Statement,) + (ir.PyAttr(slice(None)),))