mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00

Files

George Hotz 806581f807 rename rewrites + sink filter + bump to dagre 2.0.0 (#14966 )

* 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

2026-02-23 22:47:22 +08:00

1.1 KiB

Raw Blame History

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

1.1 KiB Raw Blame History