(self)
| 244 | def add_linear(self, linear:UOp, var_vals:dict[str, int]): self._linears.append(linear) |
| 245 | |
| 246 | def reset(self): |
| 247 | assert self.fxn is not None, "can't reset without function" |
| 248 | self.cnt = 0 |
| 249 | self.captured = None |
| 250 | |
| 251 | def __reduce__(self): |
| 252 | assert self.captured is not None, "can't pickle an uncaptured JIT" |
no outgoing calls