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

MlirTestUtilities

Test utilities for verifying generated MLIR correctness.

This class provides comprehensive testing utilities for validating StableHLO MLIR generation, including syntax validation, semantic checks, and round-trip validation.

Constructors

MlirTestUtilities

constructor()

Functions

createComplexTestGraph

fun createComplexTestGraph(): ComputeGraph

Create a complex test graph with multiple operation types

createSimpleTestGraph

fun createSimpleTestGraph(): ComputeGraph

Create a simple test graph for validation testing

extractPerformanceMetrics

fun extractPerformanceMetrics(module: StableHloModule): Map<String, Any>

Extract performance metrics from module generation

validateMlirPatterns

fun validateMlirPatterns(module: StableHloModule, expectedPatterns: List<String>): List<String>

Validate that generated MLIR matches expected patterns

verifyCorrectness

fun verifyCorrectness(module: StableHloModule): CorrectnessVerificationResult

Perform comprehensive correctness verification of a StableHLO module

verifyOperations

fun verifyOperations(module: StableHloModule, expectedOperations: List<String>): List<String>

Verify that a module contains expected operations

verifySSAForm

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

Verify that a module has correct SSA form

verifyTypeAnnotations

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

Verify that a module has correct type annotations