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

ScalarOperationsConverter

class ScalarOperationsConverter : StableHloOperationConverter(source)

Converter for scalar tensor operations: tensor + scalar, tensor * scalar, etc.

The KSP-generated tracing wrapper records these as ops named addScalar / subScalar / mulScalar / divScalar / rsubScalar / rdivScalar, with one tensor input and the scalar literal stored in operation.parameters["b"] (or "scalar"). The StableHLO lowering materializes the scalar as a splat constant of the output type, then applies the corresponding binary op.

rsubScalar(a, x) = a - x (scalar on the left); rdivScalar(a, x) = a / x.

Constructors

ScalarOperationsConverter

constructor()

Properties

supportedOperations

open override val supportedOperations: Set<String>

Set of operation names this converter supports

Functions

convert

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

Convert a graph node to StableHLO operations