Method _schedule_render

test/null/test_tensor.py:62–69

(self, a: Tensor)

    for src in ast.src:
      if (ret:=self._find_op(src, op)) is not None: return ret
  def _schedule_render(self, a: Tensor):
    linear, _ = a.linear_with_vars()
    for si in linear.src:
      ast = si.src[0]
      if ast.op is Ops.SINK:
        renderer = Device[si.src[1].buffer.device].renderer
        prg = to_program(ast, renderer)
        return tuple(prg.src[2].src)

  def _assert(self, dtype: DType, a: Tensor):
    uops = self._schedule_render(a)