(call:UOp, inputs:tuple[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 |
| 140 | def resolve_params(call:UOp, inputs:tuple[UOp, ...]) -> list[UOp]: return [_resolve(b, inputs) for b in get_call_arg_uops(call)] |
| 141 | |
| 142 | def unwrap_multi(call:UOp, resolved:list[UOp]) -> Iterator[tuple[list[Buffer], dict[str, int]]]: |
| 143 | bufs = [b.buffer for b in resolved] |
searching dependent graphs…