Skip to content

Instantly share code, notes, and snippets.

@jmcph4
Created January 7, 2025 23:39
Show Gist options
  • Save jmcph4/d2b889e56424c6b37fcb4dc8273dc77f to your computer and use it in GitHub Desktop.
Save jmcph4/d2b889e56424c6b37fcb4dc8273dc77f to your computer and use it in GitHub Desktop.
digraph TensorFlow {
Tensor [shape = "rect"]
LazyBuffer [shape = "rect"]
LazyOp [shape = "rect"]
UOp [shape = "rect"]
Code [shape = "rect"]
"<computation>"
Tensor -> LazyBuffer [label = "Forward and backward passes occur here (`tinygrad/function.py`)"];
LazyBuffer -> LazyOp [label = "Breaking into kernels occurs here (`tinygrad/engine/schedule.py`)"];
LazyOp -> UOp [label = "Transform kernel code into an LLVM-like IR (`tinygrad/codegen/linearize.py`)"];
UOp -> Code [label = "Transform IR into C or CUDA code (`tinygrad/renderer/*`)"];
Code -> "<computation>" [label = "Transform C or CUDA code into machine code and execute it (`tinygrad/runtime/*`)"];
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment