Skip to main content

Crate kaio_core

Crate kaio_core 

Source
Expand description

§kaio-core

PTX intermediate representation and code emission for the KAIO GPU kernel authoring framework. This crate provides the Layer 1 foundation: Rust types modelling PTX programs, instruction emitters, and a writer that renders valid .ptx text output.

§Modules

  • types — PTX type system and Rust-to-PTX type mapping
  • ir — Intermediate representation (modules, kernels, instructions)
  • instr — Instruction category enums (arithmetic, memory, control)
  • emit — PTX text emission (Emit trait + PtxWriter)
  • fragment — tensor-core fragment register containers and loaders for mma.sync / ldmatrix operands

Modules§

emit
PTX text emission — the Emit trait and PtxWriter.
fragment
Tensor-core fragment register containers for mma.sync operands.
instr
PTX instruction category enums.
ir
PTX intermediate representation — the IR tree.
types
PTX type system and Rust-to-PTX type mapping.