diff --git a/tinygrad/graph.py b/tinygrad/graph.py index 8ca7ec87f9..44ff443bfd 100644 --- a/tinygrad/graph.py +++ b/tinygrad/graph.py @@ -47,7 +47,7 @@ def log_op(ret : DeviceBuffer, ast : LazyOp, show_graph : Optional[bool] = None) op : List[Op] = [x.op for x in get_lazyops(ast)] inp : List[DeviceBuffer] = get_buffers(ast) if len(inp) == 1 and inp[0] == ret: - if nm(ret) in G.nodes: G.nodes[nm(ret)]['style'] += ', bold' + if show_graph and nm(ret) in G.nodes: G.nodes[nm(ret)]['style'] += ', bold' return # don't log self loops oporder = [LoadOps, FusedOps, ReduceOps, BinaryOps, UnaryOps, MovementOps] optype = type(sorted(op, key=lambda x: oporder.index(type(x)))[0])