Method index

tinygrad/uop/ops.py:462–463

(self, *srcs:UOp|None, ptr=False, **kwargs)

  def vectorize(self, *srcs):
    return UOp(Ops.STACK, self.dtype.vec(len(srcs)+1), (self,)+srcs)
  def index(self, *srcs:UOp|None, ptr=False, **kwargs):
    return UOp(Ops.INDEX, kwargs.pop("dtype", self.dtype if ptr else self.dtype.base), (self,)+tuple([x for x in srcs if x is not None]), **kwargs)
  def __getitem__(self, idx):
    # pointers index into INDEX UOps (scalar lookup); everything else uses the shared mixin view path
    if not isinstance(self.dtype, PtrDType): return super(UOp, self).__getitem__(idx)