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

RoundTripValidator

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

Validator for round-trip validation of StableHLO modules.

This class provides capabilities to validate that generated MLIR can be parsed and maintains semantic equivalence with the original computation graph.

Constructors

RoundTripValidator

constructor(mlirValidator: MlirValidator = MlirValidator(), parser: MlirParser = MlirParser())

Functions

checkSemanticEquivalence

fun checkSemanticEquivalence(original: StableHloModule, parsed: StableHloModule): SemanticEquivalenceReport

Check semantic equivalence between two StableHLO modules

validateParsability

fun validateParsability(module: StableHloModule): List<String>

Validate that a module can be successfully parsed and reconstructed

validateRoundTrip

fun validateRoundTrip(module: StableHloModule): RoundTripValidationResult

Perform complete round-trip validation of a StableHLO module