Function graph_class

tinygrad/engine/jit.py:88–88

(dev)

  raise JitError(f"JIT return contains non-Tensor value of type {type(ret).__name__}")

def graph_class(dev): return dev.graph.func if isinstance(dev.graph, functools.partial) else dev.graph

class GraphRunner:
  def __init__(self, linear:UOp, input_uops:tuple[UOp, ...]=()):