Module spec

Module spec 

Source

Structs§

BuiltinFusionSpec
Fusion metadata registered alongside builtin functions.
BuiltinGpuSpec
GPU metadata registered alongside builtin functions.
DocTextInventory
FusionExprContext
Context provided to fusion expression builders.
FusionKernelTemplate
Description of a fusion kernel template.
FusionSpecInventory
Inventory wrapper for fusion specs.
GpuSpecInventory
Inventory wrapper for GPU specs.

Enums§

BroadcastSemantics
Broadcast semantics supported by the builtin.
ConstantStrategy
Strategy used when embedding constants in fused kernels.
FusionError
Possible errors emitted by a fusion builder.
GpuOpKind
High-level GPU operation kind for builtin categorisation.
ProviderHook
Hook names that providers may implement for specialised kernels.
ReductionNaN
How reductions should treat NaN values.
ResidencyPolicy
Residency policy for builtin outputs.
ScalarType
Supported scalar precisions that GPU kernels may target.
ShapeRequirements
Shape requirements for fused kernels.

Functions§

builtin_doc_texts
builtin_fusion_specs
Iterate all registered fusion specs.
builtin_gpu_specs
Iterate all registered GPU specs.

Type Aliases§

FusionExprBuilder
Builder used to generate WGSL expressions.