Function disk_copy_is_buffer

tinygrad/callify.py:18–24

(ctx:AllocCtx, u:UOp)

  return x.replace(tag=(len(ctx.uop_list)-1,))

def disk_copy_is_buffer(ctx:AllocCtx, u:UOp):
  # copies to disk are replaced with the disk buffer
  to_disk = isinstance(u.device, str) and u.device.startswith(("DISK", "TINYFS"))
  if to_disk: ctx.buffer_map[u] = u.empty_like()
  # all copies from disk/numpy are realized into a real buffer
  from_creation = isinstance(u.src[0].device, str) and any(u.src[0].device.startswith(x) for x in ["NPY", "DISK", "PYTHON", "TINYFS"])
  if from_creation: return tag_uop(ctx, u)

def apply_after(ctx:AllocCtx, u:UOp):
  base = u.src[0]