Function add_to_ctx

tinygrad/function.py:8–11

(ctx, x:UOp)

from tinygrad.nn.state import get_state_dict

def add_to_ctx(ctx, x:UOp):
  ret = x.param_like(len(ctx[0]))
  ctx[0].append(x)
  return ret

pm_transform_unique_const = PatternMatcher([
  # transform unique consts to LUNIQUE