From c1bd2d3f71c81a665ceb55c1641debe13ec3ebce Mon Sep 17 00:00:00 2001
From: qazal <77887910+Qazalin@users.noreply.github.com>
Date: Fri, 1 Nov 2024 08:14:54 +0200
Subject: [PATCH] viz increment -1 kernel on enter [pr] (#7448)

* viz increment -1 kernel on enter [pr]

* two paths

* share
---
 tinygrad/viz/index.html | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/tinygrad/viz/index.html b/tinygrad/viz/index.html
index 6eb0630347..f1be9687f1 100644
--- a/tinygrad/viz/index.html
+++ b/tinygrad/viz/index.html
@@ -434,9 +434,15 @@
     }
     if (event.key == "Enter") {
       event.preventDefault()
+      if (currentKernel === -1) {
+        currentKernel = 0;
+        expandKernel = true;
+      }
+      else {
+        expandKernel = !expandKernel;
+      }
       currentUOp = 0;
       currentRewrite = 0;
-      expandKernel = !expandKernel;
       main();
     }
     if (event.key == "ArrowUp") {