From 348146870282f8a18ba34294fb78b0fbd5644ff2 Mon Sep 17 00:00:00 2001
From: qazal <77887910+Qazalin@users.noreply.github.com>
Date: Thu, 10 Oct 2024 16:56:26 +0300
Subject: [PATCH] bring viz to core (#6970)

* move viz to core

* pathfix

* move test_viz to core

* cleanup test_viz diff

* use contextvars
---
 test/test_viz.py                  | 2 +-
 tinygrad/ops.py                   | 2 +-
 {viz => tinygrad/viz}/README      | 0
 {viz => tinygrad/viz}/favicon.svg | 0
 {viz => tinygrad/viz}/index.html  | 0
 {viz => tinygrad/viz}/serve.py    | 0
 6 files changed, 2 insertions(+), 2 deletions(-)
 rename {viz => tinygrad/viz}/README (100%)
 rename {viz => tinygrad/viz}/favicon.svg (100%)
 rename {viz => tinygrad/viz}/index.html (100%)
 rename {viz => tinygrad/viz}/serve.py (100%)

diff --git a/test/test_viz.py b/test/test_viz.py
index 8296edfa8a..527c20a7cd 100644
--- a/test/test_viz.py
+++ b/test/test_viz.py
@@ -4,7 +4,7 @@ import itertools
 from tinygrad import Tensor, dtypes
 from tinygrad.helpers import Context, getenv
 from tinygrad.engine.realize import lower_schedule
-from viz.serve import GraphRewriteMetadata, get_metadata, _uop_to_json
+from tinygrad.viz.serve import GraphRewriteMetadata, get_metadata, _uop_to_json
 from tinygrad.ops import TRACK_MATCH_STATS, TrackedPatternMatcher, UPat, UOps, UOp, graph_rewrite, contexts, track_rewrites
 
 def group_rewrites(kernels:List[GraphRewriteMetadata]): return {k:list(v) for k,v in itertools.groupby(kernels, lambda x:x.loc)}
diff --git a/tinygrad/ops.py b/tinygrad/ops.py
index 1b5864f7e5..12ef986968 100644
--- a/tinygrad/ops.py
+++ b/tinygrad/ops.py
@@ -638,7 +638,7 @@ if TRACK_MATCH_STATS:
         pickle.dump(contexts, f)
     if getenv("VIZ"):
       os.environ["VIZ"] = "0"
-      os.execv(sys.executable, [sys.executable] + [os.path.join(os.path.dirname(__file__), "..", "viz", "serve.py")])
+      os.execv(sys.executable, [sys.executable] + [os.path.join(os.path.dirname(__file__), ".", "viz", "serve.py")])
     if getenv("PRINT_MATCH_STATS", 1):
       ret = [0,0,0.0,0.0]
       for k,v in sorted(list(match_stats.items()), key=lambda x: x[1][2]):
diff --git a/viz/README b/tinygrad/viz/README
similarity index 100%
rename from viz/README
rename to tinygrad/viz/README
diff --git a/viz/favicon.svg b/tinygrad/viz/favicon.svg
similarity index 100%
rename from viz/favicon.svg
rename to tinygrad/viz/favicon.svg
diff --git a/viz/index.html b/tinygrad/viz/index.html
similarity index 100%
rename from viz/index.html
rename to tinygrad/viz/index.html
diff --git a/viz/serve.py b/tinygrad/viz/serve.py
similarity index 100%
rename from viz/serve.py
rename to tinygrad/viz/serve.py