(b:UOp, inputs:tuple[UOp, ...])
| 135 | cache: bool = True |
| 136 | |
| 137 | def _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 |
| 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]]]: |
no test coverage detected
searching dependent graphs…