| 220 | return [Tensor(u) for u in UOp.custom_kernel(*[t.uop for t in (self,)+lst], fxn=fxn, grad_fxn=grad_fxn)] |
| 221 | |
| 222 | def callify(self, *lst:Tensor) -> Tensor: |
| 223 | big_sink = UOp.sink(*[x.uop for x in (self,)+lst]) |
| 224 | big_sink, buffer_map = transform_to_call(big_sink) |
| 225 | _apply_map_to_tensors({x:y.after(big_sink) for x,y in buffer_map.items()}, name="callify") |
| 226 | return self |
| 227 | |
| 228 | def linear_with_vars(self, *lst:Tensor) -> tuple[UOp, dict[str, int]]: |
| 229 | """Creates the LINEAR UOp needed to realize these Tensor(s), with Variables.""" |