| 126 | |
| 127 | @dataclass |
| 128 | class ExecContext: |
| 129 | var_vals: dict[str, int] = field(default_factory=dict) |
| 130 | input_uops: tuple[UOp, ...] = () |
| 131 | update_stats: bool = True |
| 132 | jit: bool = False |
| 133 | wait: bool = False |
| 134 | timeout: int|None = None |
| 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:])) |
no outgoing calls
no test coverage detected
searching dependent graphs…