github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 07:28:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix graph_uops (#2457)

Browse Source 
* Load networkx when we need to graph uops

* Document GRAPHUOPS

* import nx in `graph_uops`
This commit is contained in:
Yingbo Ma
2023-11-27 21:42:48 -05:00
committed by GitHub
parent 98cd9e8926
commit d43485ae9e
2 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docs/env_vars.md
View File

@@ -44,6 +44,7 @@ LLVMOPT | [1] | enable slightly more expensive LLVM optimizat
OPT | [1-3] | optimization level
BEAM | [#] | number of beams in kernel beam search
GRAPH | [1] | create a graph of all operations (requires graphviz)
GRAPHUOPS | [1] | create a graph of uops (requires graphviz and saves at /tmp/uops.{svg,dot})
GRAPHPATH | [/path/to] | where to put the generated graph
IMAGE | [1] | enable 2d specific optimizations
FLOAT16 | [1] | use float16 for images instead of float32