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

Package-level declarations

Types

data class BenchmarkResults(val testName: String, val runs: List<ConversionMetrics>, val averageMetrics: ConversionMetrics, val minMetrics: ConversionMetrics, val maxMetrics: ConversionMetrics)

Benchmark results for multiple runs

ConversionMetrics

data class ConversionMetrics(val conversionTime: Duration, val validationTime: Duration, val totalTime: Duration, val moduleSize: Int, val operationCount: Int, val ssaValueCount: Int, val memoryUsage: Long? = null)

Performance metrics for conversion operations

CorrectnessVerificationResult

data class CorrectnessVerificationResult(val isCorrect: Boolean, val syntaxValid: Boolean, val semanticValid: Boolean, val roundTripValid: Boolean, val errors: List<String> = emptyList(), val warnings: List<String> = emptyList())

Result of MLIR correctness verification

FunctionSignature

data class FunctionSignature(val parameters: List<Parameter>, val returnTypes: List<String>)

Represents a function signature in MLIR

MemoryMetrics

data class MemoryMetrics(val beforeConversion: Long, val afterConversion: Long, val estimatedUsage: Long, val moduleSize: Long)

Memory usage metrics

MlirParser

class MlirParser

Parser for MLIR StableHLO content.

MlirTestUtilities

class MlirTestUtilities

Test utilities for verifying generated MLIR correctness.

Parameter

data class Parameter(val name: String, val type: String)

Represents a function parameter

ParsedMlirStructure

data class ParsedMlirStructure(val moduleName: String? = null, val functionName: String, val functionSignature: FunctionSignature, val operations: List<ParsedOperation>, val metadata: Map<String, String> = emptyMap())

Parsed structure of an MLIR module

ParsedOperation

data class ParsedOperation(val resultName: String?, val operationType: String, val operands: List<String>, val attributes: Map<String, String> = emptyMap(), val resultType: String? = null)

Represents a parsed MLIR operation

PerformanceBenchmark

class PerformanceBenchmark

Performance benchmarking utilities for StableHLO conversion.

RoundTripValidationResult

sealed class RoundTripValidationResult

Result of round-trip validation

RoundTripValidator

class RoundTripValidator(mlirValidator: MlirValidator = MlirValidator(), parser: MlirParser = MlirParser())

Validator for round-trip validation of StableHLO modules.

SemanticEquivalenceReport

data class SemanticEquivalenceReport(val isEquivalent: Boolean, val functionSignatureMatch: Boolean, val operationCountMatch: Boolean, val ssaStructureMatch: Boolean, val typeConsistency: Boolean, val differences: List<String> = emptyList())

Report on semantic equivalence between original and parsed modules

TestScenario

data class TestScenario(val name: String, val graph: ComputeGraph, val expectedOperations: List<String> = emptyList(), val expectedPatterns: List<String> = emptyList())

Represents a test scenario for MLIR validation

TestScenarioBuilder

class TestScenarioBuilder

Builder for creating test scenarios

ThroughputMetrics

data class ThroughputMetrics(val conversionsCompleted: Int, val totalOperations: Int, val actualTime: Duration, val conversionsPerSecond: Double, val operationsPerSecond: Double)

Throughput metrics for conversion operations

ValidationStage

enum ValidationStage : Enum<ValidationStage>

Stages of validation where failures can occur