Method new_buffer

tinygrad/uop/ops.py:710–711

(device:str|tuple[str, ...], size:int, dtype:DType, num=None)

  @staticmethod
  def new_buffer(device:str|tuple[str, ...], size:int, dtype:DType, num=None):
    return UOp(Ops.BUFFER, dtype, (UOp.unique(num), UOp(Ops.DEVICE, arg=device)), size)
  @staticmethod
  def from_buffer(opaque:Buffer, device:str|tuple[str, ...]|None=None):
    buffers[uop:=UOp.new_buffer(device or opaque.device, opaque.size, opaque.dtype)] = opaque.ref(1)