diff --git a/tinygrad/uop/ops.py b/tinygrad/uop/ops.py index 1ea85c5766..4826ddab2c 100644 --- a/tinygrad/uop/ops.py +++ b/tinygrad/uop/ops.py @@ -1422,6 +1422,7 @@ def bitcast(x, in_dtype:DType, out_dtype:DType): renderer = PatternMatcher([ (UPat((Ops.DEFINE_VAR,), name="x"), lambda x: x.expr), + (UPat(Ops.PARAM, src=(UPat(), UPat(), UPat(), UPat(), UPat(Ops.NOOP, name="x"))), lambda x: x.arg), (UPat((Ops.SPECIAL), name="x"), lambda x: x.arg), (UPat(Ops.RANGE, name="x"), lambda x: f"r{range_str(x)}"), (UPat((Ops.CONST, Ops.VCONST), name="x"), lambda x: str(x.arg)), diff --git a/tinygrad/viz/serve.py b/tinygrad/viz/serve.py index 0e3cc801bb..4aac563d3b 100755 --- a/tinygrad/viz/serve.py +++ b/tinygrad/viz/serve.py @@ -128,6 +128,8 @@ def uop_to_json(x:UOp) -> dict[int, dict]: label += f"\n({multirange_str(rngs, color=True)})" if u._shape is not None: label += f"\n{shape_to_str(u.shape)}" + if u.op is Ops.CALL: + label += f"\n{u.src[0].key.hex()[:8]}" if u.op in {Ops.INDEX, Ops.BUFFERIZE}: if len(u.toposort()) < 30: label += f"\n{u.render()}" ranges: list[UOp] = []