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

BasicExample

object BasicExample(source)

Basic example demonstrating the new StableHLO converter architecture.

This example shows how to:

Create a computational graph
Use the new converter to generate StableHLO MLIR
Validate the output

Functions

convertToStableHlo

fun convertToStableHlo(graph: DefaultComputeGraph): String

Convert the graph to StableHLO MLIR using the new converter

createSimpleGraph

fun createSimpleGraph(): DefaultComputeGraph

Create a simple graph: input -> add -> relu

fun runExample(): String

Run the complete example