From 8bc9277587a07bb66641cbccdee85182cea80450 Mon Sep 17 00:00:00 2001
From: George Hotz <geohot@gmail.com>
Date: Sat, 4 Mar 2023 07:24:43 -0800
Subject: [PATCH] G.nodes isn't always valid

---
 tinygrad/graph.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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])