skainet-compile-hlo/sk.ainet.compile.hlo.converters/LinalgOperationsConverter

LinalgOperationsConverter

class LinalgOperationsConverter : StableHloOperationConverter(source)

Converter for linear algebra operations.

This converter implements support for matrix operations including matrix multiplication and transpose operations as specified in Requirements 2.2:

Matrix multiplication (matmul) using stablehlo.dot_general
Transpose operations with arbitrary dimension permutations
Batch matrix operations support
Proper dot_general configuration for contracting dimensions

The converter handles:

2D matrix multiplication (standard matmul)
Batch matrix multiplication (3D+ tensors)
Transpose with configurable dimension permutations
Proper type inference and shape handling

Constructors

LinalgOperationsConverter

constructor()

Properties

supportedOperations

open override val supportedOperations: Set<String>

Set of operation names this converter supports

Functions

open override fun convert(node: GraphNode, operands: List<String>, context: ConversionContext): ConversionResult

Convert a graph node to StableHLO operations