From 972d9ea215f65922cf86bfda07bea920c198b60b Mon Sep 17 00:00:00 2001 From: Roelof van Dijk <3604013+roelofvandijk@users.noreply.github.com> Date: Thu, 5 Oct 2023 23:28:43 +0200 Subject: [PATCH] fix: PRUNEGRAPH is unused (#1985) --- docs/env_vars.md | 1 - tinygrad/helpers.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/docs/env_vars.md b/docs/env_vars.md index 06c27dac6a..c6d0afbfe5 100644 --- a/docs/env_vars.md +++ b/docs/env_vars.md @@ -47,7 +47,6 @@ KOPT | [1-2] | kernel optimization, 1 turns it on, 2 caches BUDGET | [#] | kernel optimization search budget GRAPH | [1] | create a graph of all operations (requires graphviz) GRAPHPATH | [/path/to] | where to put the generated graph -PRUNEGRAPH | [1] | prune MovementOps and LoadOps from the graph PRINT_PRG | [1] | print program code IMAGE | [1] | enable 2d specific optimizations FLOAT16 | [1] | use float16 for images instead of float32 diff --git a/tinygrad/helpers.py b/tinygrad/helpers.py index 88ce1114d9..2200f9db32 100644 --- a/tinygrad/helpers.py +++ b/tinygrad/helpers.py @@ -60,7 +60,7 @@ class ContextVar: def __lt__(self, x): return self.value < x DEBUG, IMAGE = ContextVar("DEBUG", 0), ContextVar("IMAGE", 0) -GRAPH, PRUNEGRAPH, GRAPHPATH = getenv("GRAPH", 0), getenv("PRUNEGRAPH", 0), getenv("GRAPHPATH", "/tmp/net") +GRAPH, GRAPHPATH = getenv("GRAPH", 0), getenv("GRAPHPATH", "/tmp/net") class Timing(contextlib.ContextDecorator): def __init__(self, prefix="", on_exit=None, enabled=True): self.prefix, self.on_exit, self.enabled = prefix, on_exit, enabled