mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-07-17 07:12:08 +08:00
firestar5683
42 lines
1.7 KiB
Python
42 lines
1.7 KiB
Python
from __future__ import annotations
|
|
import itertools
|
|
from dataclasses import dataclass, field
|
|
from tinygrad.renderer import Renderer
|
|
from tinygrad.uop.ops import PatternMatcher, UOp, Ops, consumer_map_from_toposort
|
|
|
|
@dataclass(frozen=True)
|
|
class Register:
|
|
name: str
|
|
index: int
|
|
_cons: tuple[Register, ...] = field(default_factory=tuple)
|
|
@property
|
|
def cons(self): return self._cons or (self,)
|
|
def __repr__(self): return self.name
|
|
|
|
class IselContext:
|
|
def __init__(self, sink:UOp):
|
|
self.uses = consumer_map_from_toposort(sink.toposort())
|
|
self.reg_n = itertools.count()
|
|
arg_order = {Ops.PARAM: 0, Ops.SPECIAL: 1}
|
|
self.func_args = sorted([u for u in self.uses if u.op in arg_order], key=lambda k: (arg_order[k.op], k.arg))
|
|
|
|
def vreg(self, cons:tuple[Register, ...]|Register):
|
|
return Register(f"v{next(self.reg_n)}", 0, _cons=cons if isinstance(cons, tuple) else (cons,))
|
|
|
|
@dataclass
|
|
class PreRegAllocContext:
|
|
lock: UOp|None = None
|
|
clobbered: set[UOp] = field(default_factory=set)
|
|
|
|
class ISARenderer(Renderer):
|
|
pre_isel_matcher: PatternMatcher
|
|
isel_matcher: PatternMatcher
|
|
pre_regalloc_matcher: PatternMatcher|None = None
|
|
post_regalloc_matcher: PatternMatcher
|
|
|
|
def is_two_address(self, x:UOp) -> bool: return False
|
|
def stack_pointer(self) -> UOp: raise NotImplementedError("arch specific")
|
|
def copy(self, x:UOp, reg:Register) -> UOp: raise NotImplementedError("arch specific")
|
|
def spill(self, disp:UOp, x:UOp) -> UOp: raise NotImplementedError("arch specific")
|
|
def fill(self, disp:UOp, x:UOp, reg:Register) -> UOp: raise NotImplementedError("arch specific")
|
|
def asm_str(self, uops:list[UOp], function_name:str) -> str: raise NotImplementedError("arch specific") |