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

Package-level declarations

Types

class ConstantFoldingPass : OptimizationPass

Constant folding optimization pass

sealed interface ConstantMaterializationPolicy

Policy governing how a constant tensor (a weight, a bias, a frozen parameter) is materialized into the emitted StableHLO text.

ConversionContext

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

Context object for maintaining state during StableHLO conversion.

ConversionResult

sealed class ConversionResult

Result of an operation conversion

DeadCodeEliminationPass

class DeadCodeEliminationPass : OptimizationPass

Dead code elimination pass

ExternalParameterRef

data class ExternalParameterRef(val scope: String, val key: String, val encoding: TensorEncoding, val source: BufferHandle)

Reference to a weight tensor that has been lifted out of the emitted StableHLO text and moved behind an util.global.load reference. The converter produces these; a downstream packager consumes them to write an IREE parameter archive (.irpa).

MlirValidator

class MlirValidator

Validator for MLIR syntax and semantic correctness.

OperationFusionPass

class OperationFusionPass : OptimizationPass

Operation fusion optimization pass

OptimizationPass

interface OptimizationPass

Interface for optimization passes

RegistryStats

data class RegistryStats(val converterCount: Int, val operationCount: Int)

Statistics about the operation registry

StableHloConverter

class StableHloConverter @JvmOverloads constructor(registry: StableHloOperationRegistry, typeMapper: TypeMapper, validator: MlirValidator? = null, materializationPolicy: ConstantMaterializationPolicy = ConstantMaterializationPolicy.InlineAlways)

Main converter class that orchestrates the conversion process from ComputeGraph to StableHLO MLIR.

StableHloConverterFactory

object StableHloConverterFactory

Factory for creating StableHLO converters with default configurations.

StableHloModule

data class StableHloModule(val content: String, val functionName: String = "main", val inputSpecs: List<TensorSpec> = emptyList(), val outputSpecs: List<TensorSpec> = emptyList(), val metadata: Map<String, Any> = emptyMap(), val externalParameters: List<ExternalParameterRef> = emptyList())

Represents a StableHLO MLIR module text output

StableHloOperationConverter

interface StableHloOperationConverter

Interface for converting specific operations to StableHLO format.

StableHloOperationRegistry

class StableHloOperationRegistry

Registry for StableHLO operation converters.

StableHloOptimizer

class StableHloOptimizer

Optimizer for StableHLO MLIR modules.

TypeMapper

class TypeMapper