Index

A

acc (compiler.lowering.arithmetic.ArithResult attribute) acc_dtype (compiler.ir.KernelIR attribute) addr (compiler.ir.IRBufLoad attribute) (compiler.ir.IRBufStore attribute)

amaranth_shape() (compiler.ir.DType method) analyze_buffers() (in module compiler.backend) ArithmeticLowering (class in compiler.lowering.arithmetic) ArithResult (class in compiler.lowering.arithmetic) axis_type (compiler.ir.LoopIR attribute)

B

BF16 (compiler.ir.DType attribute) body (compiler.ir.LoopIR attribute) bound (compiler.ir.IRCounter attribute) (compiler.ir.LoopIR attribute) buf_idx (compiler.ir.IRBufLoad attribute) (compiler.ir.IRBufStore attribute)

buf_infos (compiler.backend.KernelSpec attribute) buf_map (compiler.backend.KernelSpec attribute) BufferInfo (class in compiler.backend) BufferMeta (class in compiler.ir) buffers (compiler.ir.KernelIR attribute) build_control() (in module compiler.lowering.control)

C

compile_kernel() (in module compiler.backend) compile_model() (in module compiler.backend) compile_top_module() (in module compiler.backend) CompiledKernel (class in compiler.hdl_module) compiler module compiler.backend module compiler.fp32 module compiler.hdl_module module compiler.ir module

compiler.lowering.arithmetic module compiler.lowering.control module compiler.top_module module compiler.uop_to_ir module compiler.utils module const_to_bits() (compiler.ir.DType method) count_cycles_from_schedule() (in module compiler.backend) counter_sigs (compiler.lowering.arithmetic.ArithResult attribute) create_counters() (in module compiler.lowering.arithmetic)

D

depth (compiler.backend.BufferInfo attribute) (compiler.ir.BufferMeta attribute) (compiler.ir.IRCounter attribute) (compiler.ir.LoopIR attribute) device (compiler.backend.HDLRenderer attribute)

DType (class in compiler.ir) dtype (compiler.ir.BufferMeta attribute) (compiler.ir.IRBufStore attribute) (compiler.ir.IRConst attribute) (compiler.ir.IROp attribute) (compiler.ir.IRRegLoad attribute)

E

elaborate() (compiler.fp32.FP32Add method) (compiler.fp32.FP32Cmp method) (compiler.fp32.FP32Exp2 method) (compiler.fp32.FP32FDiv method) (compiler.fp32.FP32Log2 method) (compiler.fp32.FP32Mul method) (compiler.fp32.FP32Reciprocal method) (compiler.fp32.FP32Sqrt method) (compiler.hdl_module.CompiledKernel method) (compiler.top_module.TopModule method)

elem_width (compiler.backend.BufferInfo attribute) epilogue (compiler.ir.LoopIR attribute)

F

format() (compiler.ir.KernelIR method) format_uops() (in module compiler.utils) FP16 (compiler.ir.DType attribute) FP32 (compiler.ir.DType attribute) FP32Add (class in compiler.fp32) FP32Cmp (class in compiler.fp32) FP32Exp2 (class in compiler.fp32)

FP32FDiv (class in compiler.fp32) FP32Log2 (class in compiler.fp32) FP32Mul (class in compiler.fp32) FP32Reciprocal (class in compiler.fp32) FP32Sqrt (class in compiler.fp32) from_tinygrad() (compiler.ir.DType class method) from_width() (compiler.ir.DType class method)

G

global_max (compiler.backend.HDLRenderer attribute)

H

has_local (compiler.backend.HDLRenderer attribute)

has_shared (compiler.backend.HDLRenderer attribute) HDLRenderer (class in compiler.backend)

I

idx (compiler.backend.BufferInfo attribute) (compiler.ir.BufferMeta attribute) INT16 (compiler.ir.DType attribute) INT32 (compiler.ir.DType attribute) INT8 (compiler.ir.DType attribute) IRBufLoad (class in compiler.ir) IRBufStore (class in compiler.ir)

IRConst (class in compiler.ir) IRCounter (class in compiler.ir) IROp (class in compiler.ir) IRRegLoad (class in compiler.ir) IRRegStore (class in compiler.ir) is_output (compiler.backend.BufferInfo attribute) (compiler.ir.BufferMeta attribute) is_signed (compiler.backend.BufferInfo attribute)

K

kernel (compiler.backend.KernelSpec attribute)

KernelIR (class in compiler.ir) KernelSpec (class in compiler.backend)

L

local_max (compiler.backend.HDLRenderer attribute)

loop_tree (compiler.ir.KernelIR attribute) LoopIR (class in compiler.ir)

M

module compiler compiler.backend compiler.fp32 compiler.hdl_module compiler.ir compiler.lowering.arithmetic compiler.lowering.control compiler.top_module compiler.uop_to_ir compiler.utils

O

op (compiler.ir.IROp attribute)

P

pretty() (compiler.ir.KernelIR method)

pretty_print_uops() (in module compiler.utils) prologue (compiler.ir.LoopIR attribute)

R

render() (compiler.backend.HDLRenderer method)

run() (compiler.lowering.arithmetic.ArithmeticLowering method)

S

scalar_stores (compiler.ir.KernelIR attribute) show_hardware() (in module compiler.utils) signals (compiler.lowering.arithmetic.ArithResult attribute) simulate_kernel() (in module compiler.backend)

simulate_top() (in module compiler.top_module) srcs (compiler.ir.IROp attribute) supports_float4 (compiler.backend.HDLRenderer attribute) synthesis_stats() (in module compiler.utils)

T

TopModule (class in compiler.top_module)

U

UINT16 (compiler.ir.DType attribute) UINT32 (compiler.ir.DType attribute) UINT8 (compiler.ir.DType attribute)

uop_to_ir() (in module compiler.uop_to_ir) uops (compiler.backend.KernelSpec attribute) uops_to_kernel_ir() (in module compiler.backend)

V

value (compiler.ir.IRBufStore attribute) (compiler.ir.IRConst attribute) (compiler.ir.IRRegStore attribute)