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

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())(source)

Represents a StableHLO MLIR module text output

Constructors

StableHloModule

constructor(content: String, functionName: String = "main", inputSpecs: List<TensorSpec> = emptyList(), outputSpecs: List<TensorSpec> = emptyList(), metadata: Map<String, Any> = emptyMap())

Properties

val content: String

val functionName: String

val inputSpecs: List<TensorSpec>

val metadata: Map<String, Any>

val outputSpecs: List<TensorSpec>

Functions

fun StableHloModule.asInputStream(): InputStream

Returns an InputStream over the MLIR content encoded as UTF-8.

fun StableHloModule.asReader(): Reader

Returns a Reader over the MLIR content.

fun contentLines(): Sequence<String>

Lazily iterate over the MLIR content line by line.

fun optimize(optimizer: StableHloOptimizer): StableHloModule

Optimize this module using the provided optimizer

fun validate(validator: MlirValidator): List<String>

Validate this module using the provided validator