Remove SymbolTable and WorkList from codegen classes QASM and STIM. Code generation base EmitABC now contains them.

Author: zhenrongliew

src/bloqade/qasm2/emit/base.py

@@ -82,31 +82,3 @@ def assert_node(
         if not isinstance(node, typ):
             raise TypeError(f"expected {typ}, got {type(node)}")
         return node
-
-
-@dataclass
-class SymbolTable(idtable.IdTable[ir.Statement]):
-    def add(self, value: ir.Statement) -> str:
-        id = self.next_id
-        if (trait := value.get_trait(ir.SymbolOpInterface)) is not None:
-            value_name = trait.get_sym_name(value).unwrap()
-            curr_ind = self.name_count.get(value_name, 0)
-            suffix = f"_{curr_ind}" if curr_ind != 0 else ""
-            self.name_count[value_name] = curr_ind + 1
-            name = self.prefix + value_name + suffix
-            self.table[value] = name
-        else:
-            name = f"{self.prefix}{self.prefix_if_none}{id}"
-            self.next_id += 1
-            self.table[value] = name
-        return name
-
-    def __getitem__(self, value: ir.Statement) -> str:
-        if value in self.table:
-            return self.table[value]
-        raise KeyError(f"Symbol {value} not found in SymbolTable")
-
-    def get(self, value: ir.Statement, default: str | None = None) -> str | None:
-        if value in self.table:
-            return self.table[value]
-        return default

src/bloqade/qasm2/emit/gate.py

@@ -2,14 +2,13 @@
 
 from kirin import ir, types, interp
 from kirin.dialects import py, func, ilist
-from kirin.worklist import WorkList
 from kirin.ir.dialect import Dialect as Dialect
 from typing_extensions import Self
 
 from bloqade.types import QubitType
 from bloqade.qasm2.parse import ast
 
-from .base import SymbolTable, EmitQASM2Base, EmitQASM2Frame
+from .base import EmitQASM2Base, EmitQASM2Frame
 
 
 def _default_dialect_group():
@@ -25,9 +24,6 @@ class EmitQASM2Gate(EmitQASM2Base[ast.UOp | ast.Barrier, ast.Gate]):
 
     def initialize(self) -> Self:
         super().initialize()
-        if getattr(self, "callables", None) is None:
-            self.callables = SymbolTable(prefix="")
-        self.callable_to_emit = WorkList()
         return self
 
     def run(self, node: ir.Method | ir.Statement):

src/bloqade/qasm2/emit/main.py

@@ -1,30 +1,26 @@
 from typing import List, cast
-from dataclasses import field, dataclass
+from dataclasses import dataclass
 
 from kirin import ir, interp
 from kirin.dialects import cf, scf, func
-from kirin.worklist import WorkList
 from kirin.ir.dialect import Dialect as Dialect
 from typing_extensions import Self
 
 from bloqade.qasm2.parse import ast
 from bloqade.qasm2.dialects.uop import SingleQubitGate, TwoQubitCtrlGate
 from bloqade.qasm2.dialects.expr import GateFunction
 
-from .base import SymbolTable, EmitQASM2Base, EmitQASM2Frame
+from .base import EmitQASM2Base, EmitQASM2Frame
 from ..dialects.core.stmts import Reset, Measure
 
 
 @dataclass
 class EmitQASM2Main(EmitQASM2Base[ast.Statement, ast.MainProgram]):
     keys = ("emit.qasm2.main", "emit.qasm2.gate")
     dialects: ir.DialectGroup
-    callable_to_emit: WorkList[ir.Statement] = field(init=False)
 
     def initialize(self) -> Self:
         super().initialize()
-        self.callables: SymbolTable = SymbolTable(prefix="")
-        self.callable_to_emit = WorkList()
         return self
 
     def run(self, node: ir.Method | ir.Statement):

src/bloqade/stim/emit/stim_str.py

@@ -7,7 +7,6 @@
 from kirin.idtable import IdTable
 from kirin.dialects import func
 from kirin.emit.abc import EmitABC, EmitFrame
-from kirin.worklist import WorkList
 
 IO_t = TypeVar("IO_t", bound=IO)
 
@@ -43,14 +42,10 @@ class EmitStimMain(EmitABC[EmitStimFrame, str], Generic[IO_t]):
     io: IO_t = cast(IO_t, sys.stdout)
     keys = ("emit.stim",)
     void = ""
-    callables: IdTable[ir.Statement] = field(init=False)
-    callable_to_emit: WorkList[ir.Statement] = field(init=False)
     correlation_identifier_offset: int = 0
 
     def initialize(self) -> "EmitStimMain":
         super().initialize()
-        self.callables: IdTable[ir.Statement] = IdTable(prefix="fn_")
-        self.callable_to_emit: WorkList[ir.Statement] = WorkList()
         self.correlated_error_count = self.correlation_identifier_offset
         return self