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

BasicMathConverter

class BasicMathConverter : StableHloOperationConverter(source)

Converter for basic mathematical operations (add, subtract, multiply, divide).

This converter handles the basic arithmetic operations by mapping them to their corresponding StableHLO operations. It supports:

Element-wise operations with broadcasting
Mixed-type arithmetic with automatic type promotion
Proper operand ordering and type consistency

Constructors

BasicMathConverter

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