geohot/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Actions Activity

fix: PRUNEGRAPH is unused (#1985)

Browse Source
This commit is contained in:
Roelof van Dijk
2023-10-05 14:28:43 -07:00
committed by GitHub
parent 21a2c5df73
commit 972d9ea215
2 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/env_vars.md
-1
View File
@@ -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
tinygrad/helpers.py
+1 -1
View File
@@ -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