MCPcopy
hub / github.com/tinygrad/tinygrad / _resolve

Function _resolve

tinygrad/engine/realize.py:137–139  ·  view source on GitHub ↗
(b:UOp, inputs:tuple[UOp, ...])

Source from the content-addressed store, hash-verified

135 cache: bool = True
136
137def _resolve(b:UOp, inputs:tuple[UOp, ...]) -> UOp:
138 if b.op in (Ops.BUFFER_VIEW, Ops.MSELECT) and b.src[0].op is Ops.PARAM: return b.replace(src=(inputs[b.src[0].arg], *b.src[1:]))
139 return inputs[b.arg] if b.op is Ops.PARAM else b
140def resolve_params(call:UOp, inputs:tuple[UOp, ...]) -> list[UOp]: return [_resolve(b, inputs) for b in get_call_arg_uops(call)]
141
142def unwrap_multi(call:UOp, resolved:list[UOp]) -> Iterator[tuple[list[Buffer], dict[str, int]]]:

Callers 1

resolve_paramsFunction · 0.85

Calls 1

replaceMethod · 0.45

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…