github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

mypy

Browse Source
This commit is contained in:
Christopher Milan
2025-12-18 21:05:46 +00:00
parent 7462edaffe
commit 68502a0e88
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tinygrad/renderer/__init__.py
View File

@@ -64,7 +64,7 @@ class ProgramSpec:
device:str
ast:UOp # save the base ast (this is method cache key)
uops:list[UOp]|None=None
aux:dict[str,...]=field(default_factory=dict)
aux:dict=field(default_factory=dict)
# filled in from uops (if we have uops)
global_size:list[int]|None=None
@@ -133,5 +133,5 @@ class Renderer:
code_for_op: dict[Ops, Callable] = {}
def __reduce__(self): return self.__class__, ()
def render(self, uops:list[UOp]) -> str|tuple[str,...]: raise NotImplementedError("needs a renderer")
def aux(self, uops:list[UOp]) -> dict[str,...]: raise NotImplementedError("needs aux")
def render(self, uops:list[UOp]) -> str: raise NotImplementedError("needs a renderer")
def aux(self, uops:list[UOp]) -> dict: raise NotImplementedError("needs aux")