Module Tolk

module Allocations : sig ... end

Buffer allocation for tensor graphs.

module Allreduce : sig ... end

Multi-device collective reduction.

module Coalese : sig ... end

Late memory coalescing and image selection.

module Codegen : sig ... end

Codegen entry point — optimization dispatch, lowering, and compilation.

module Codegen_lower : sig ... end

Codegen lowering after optimization.

module Compiler : sig ... end

Kernel source compiler.

module Cstyle : sig ... end

C-family language renderers.

module Decomp_dtype : sig ... end

Dtype decomposition matchers matching tinygrad codegen/decomp/dtype.py.

module Decomp_op : sig ... end

Operation decompositions matching tinygrad codegen/decomp/op.py.

module Decomp_transcendental : sig ... end

Transcendental decompositions matching tinygrad codegen/decomp/transcendental.py.

module Device : sig ... end

Device runtime abstraction.

module Diskcache : sig ... end

Simple file-based disk cache.

module Elf : sig ... end

Relocatable ELF object loading.

module Gater : sig ... end

Late gate relocation.

module Gpu_dim : sig ... end

Canonical GPU workitem dimensions.

module Gpu_target : sig ... end

Resolved GPU target descriptors for renderer construction.

module Gpudims : sig ... end

GPU dimension mapping.

module Helpers : sig ... end
module Heuristic : sig ... end

Heuristic-based kernel optimizations.

module Indexing : sig ... end

Rangeify: tensor graph to indexed representation.

module Jit : sig ... end

JIT compilation and replay.

module Linearizer : sig ... end

Kernel-to-program linearization.

module Multi : sig ... end

Multi-device sharding transformations.

module Postrange : sig ... end

Post-range kernel optimisation scheduler.

module Program_spec : sig ... end

Compile-time kernel descriptions extracted from a linearized uop program.

module Rangeify : sig ... end

Schedule pipeline: tensor graph to kernel graph.

module Realize : sig ... end

Linear execution and kernel dispatch.

module Renderer : sig ... end

GPU kernel renderer.

module Schedule : sig ... end

Tensor schedule linearization.

Beam search kernel optimiser.

module Simplify : sig ... end

Range simplification passes.

module Tc : sig ... end

Tinygrad tensor core definitions and swizzle helpers.

module Tlsf : sig ... end

Two-Level Segregated Fit allocator.