skainet-compile-hlo/sk.ainet.compile.hlo/ConversionContext

ConversionContext

class ConversionContext @JvmOverloads constructor(typeMapper: TypeMapper, graph: ComputeGraph? = null, val materializationPolicy: ConstantMaterializationPolicy = ConstantMaterializationPolicy.InlineAlways)(source)

Context object for maintaining state during StableHLO conversion.

This class manages SSA value names, type mapping, and MLIR code generation during the conversion process from ComputeGraph to StableHLO.

Constructors

ConversionContext

@JvmOverloads

constructor(typeMapper: TypeMapper, graph: ComputeGraph? = null, materializationPolicy: ConstantMaterializationPolicy = ConstantMaterializationPolicy.InlineAlways)

Properties

materializationPolicy

val materializationPolicy: ConstantMaterializationPolicy

Governs whether constant tensors are inlined as dense<...> or lifted into util.global module declarations. Default ConstantMaterializationPolicy.InlineAlways preserves historical behavior for every caller that constructs a context without naming a policy — the external path is strictly opt-in. See issue #523 for the architecture context.

Functions

clear

fun clear()

Clear all state (useful for testing)

emitComment

fun emitComment(comment: String)

Emit a comment with proper indentation

emitEncodingAnnotation

fun emitEncodingAnnotation(role: String, index: Int, spec: TensorSpec)

Emit a tensor_encoding diagnostic comment when spec carries a non-null tensorEncoding (set via sk.ainet.lang.tensor.ops.withTensorEncoding).

emitLine

fun emitLine(line: String)

Emit a line of MLIR code with proper indentation

emitModuleDeclaration

fun emitModuleDeclaration(line: String)

Emit a module-scope declaration (e.g. util.global private @w : ...).

emitOperation

fun emitOperation(operation: String)

Emit an operation with proper indentation

getContent

fun getContent(): String

Get the complete generated MLIR content

getExternalParameters

fun getExternalParameters(): List<ExternalParameterRef>

Snapshot of every externalized constant registered during this conversion. Surfaced on StableHloModule.externalParameters.

getInputNodes

fun getInputNodes(node: GraphNode): List<GraphNode>

Get input nodes for a given node from the graph

getModuleDeclarations

fun getModuleDeclarations(): String

Return every module-scope declaration emitted so far. Used by StableHloConverter when assembling the final content.

getTypeMapper

fun getTypeMapper(): TypeMapper

Get the type mapper instance

getValueName

fun getValueName(nodeId: String): String?

Get the SSA value name for a node ID

getValueType

fun getValueType(valueName: String): String?

Get the MLIR tensor type for an SSA value name, or null if the value was produced by a converter that did not record its type.

nextTempValue

fun nextTempValue(): String

Generate the next temporary SSA value name

registerExternalParameter

fun registerExternalParameter(ref: ExternalParameterRef)

Register an externalized constant tensor. The converter records these alongside MLIR emission so a downstream packager (see PR C in issue #523) can write them into an IREE .irpa archive.

setGraph

fun setGraph(graph: ComputeGraph)

Set the graph reference for node lookups

setValueName

fun setValueName(nodeId: String, valueName: String)

Set the SSA value name for a node ID

setValueType

fun setValueType(valueName: String, mlirType: String)

Record the MLIR tensor type associated with an SSA value name.