mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-04-29 03:00:14 -04:00
806581f807
* bump to dagre 2.0.0 * transform to call * cleanup names * get kernel graph * dagre recursion fix + better error * add toggle to hide sink nodes * no sink by default * revert that * only hide final sinks * lol
61 lines
1.1 KiB
Markdown
61 lines
1.1 KiB
Markdown
# tinygrad directory layout
|
|
|
|
This explains the flow of a big graph down to programs.
|
|
|
|
Directories are listed in order of how they are processed.
|
|
|
|
---
|
|
|
|
## tinygrad/schedule
|
|
|
|
Group UOps into kernels.
|
|
|
|
::: tinygrad.schedule.rangeify.get_kernel_graph
|
|
options:
|
|
members: false
|
|
show_labels: false
|
|
show_source: false
|
|
|
|
---
|
|
|
|
## tinygrad/codegen/opt
|
|
|
|
Transforms the ast into an optimized ast. This is where BEAM search and heuristics live.
|
|
|
|
---
|
|
|
|
## tinygrad/codegen
|
|
|
|
Transform the optimized ast into a linearized and rendered program.
|
|
|
|
::: tinygrad.codegen.get_program
|
|
options:
|
|
members: false
|
|
show_labels: false
|
|
show_source: false
|
|
|
|
---
|
|
|
|
## tinygrad/renderer
|
|
|
|
Transform the linearized list of UOps into a program, represented as a string.
|
|
|
|
::: tinygrad.renderer.Renderer
|
|
options:
|
|
members:
|
|
- render
|
|
show_labels: false
|
|
show_source: false
|
|
|
|
---
|
|
|
|
## tinygrad/engine
|
|
|
|
Abstracted high level interface to the runtimes.
|
|
|
|
::: tinygrad.engine.realize.get_program
|
|
options:
|
|
members: false
|
|
show_labels: false
|
|
show_source: false
|