FHElium

Appearance

fhelium.backend.triton._expressions ​

Component expression graphs and indexed loads for Triton code generation.

ARITHMETIC_OPS ​

constant View source

python
ARITHMETIC_OPS = (ckks.AddCompressedPlaintextOp, ckks.MultiplyCompressedPlaintextOp, rns.AddStandardOp, rns.SubtractStandardOp, rns.NegateStandardOp, rns.MontgomeryMultiplyOp, rns.AddMontgomeryLazyOp, rns.MultiplyPlaintextOp, rns.AddPlaintextOp, rns.StandardToMontgomeryOp, rns.MontgomeryToStandardOp)

STRUCTURAL_OPS ​

constant View source

python
STRUCTURAL_OPS = (rns.ExtractComponentOp, rns.PackTwoComponentsOp, rns.PackThreeComponentsOp)

NTT_OPS ​

constant View source

python
NTT_OPS = (ntt.CoefficientStandardToNttMontgomeryOp, ntt.CoefficientMontgomeryToNttMontgomeryOp, ntt.NttMontgomeryToCoefficientStandardOp, ntt.NttMontgomeryToCoefficientMontgomeryOp)

SUPPORTED_OPERATIONS ​

constant View source

python
SUPPORTED_OPERATIONS = ARITHMETIC_OPS + STRUCTURAL_OPS + (rns.KeySwitchDigitProductOp,)

Expr ​

class View source

python
Expr(op: str, args: tuple[Expr, ...] = (), data: tuple[int, ...] = ())

Attributes ​

NameTypeDefault/value
opstr
argstuple[Expr, ...]()
datatuple[int, ...]()

InputLayout ​

class View source

python
InputLayout(shape: tuple[int, ...], strides: tuple[int, ...], component_axis: bool = False)

Attributes ​

NameTypeDefault/value
shapetuple[int, ...]
stridestuple[int, ...]
component_axisboolFalse

components ​

property

python
components: int

batch_shape ​

property

python
batch_shape: tuple[int, ...]

ValueExpr ​

class View source

python
ValueExpr(components: tuple[Expr, ...], shape: tuple[int, ...], component_axis: bool, root: int, view: tuple[int, ...] = ())

Attributes ​

NameTypeDefault/value
componentstuple[Expr, ...]
shapetuple[int, ...]
component_axisbool
rootint
viewtuple[int, ...]()

batch_shape ​

property

python
batch_shape: tuple[int, ...]

Anchor ​

class View source

python
Anchor(operation: Operation, source: ValueExpr, result: ValueExpr, table_indices: tuple[int, ...])

Attributes ​

NameTypeDefault/value
operationOperation
sourceValueExpr
resultValueExpr
table_indicestuple[int, ...]

BoundGraph ​

class View source

python
BoundGraph(layouts: tuple[InputLayout, ...], outputs: tuple[ValueExpr, ...], roots: Mapping[int, ValueExpr], anchors: tuple[Anchor, ...] = ())

Attributes ​

NameTypeDefault/value
layoutstuple[InputLayout, ...]
outputstuple[ValueExpr, ...]
rootsMapping[int, ValueExpr]
anchorstuple[Anchor, ...]()

batch_count ​

property

python
batch_count: int

component_axis ​

function View source

python
def component_axis(value_type: object) -> bool: ...

ExpressionGraph ​

class View source

python
ExpressionGraph(operation: fusion.FusedOp)

Bind a region's SSA DAG to input shapes without specializing data values.

Attributes ​

NameTypeDefault/value
operationfusion.FusedOp

table_arguments ​

property

python
table_arguments: tuple[SSAValue, ...]

tensor_types ​

property

python
tensor_types: tuple[object, ...]

input_indices ​

property

python
input_indices: tuple[tuple[int, ...], tuple[int, ...]]

parameter_index ​

property

python
parameter_index: int

key_inputs ​

property

python
key_inputs: tuple[tuple[int, int, int], ...]

bind ​

method

python
def bind(layouts: tuple[InputLayout, ...]) -> BoundGraph: ...

compact_value ​

function View source

python
def compact_value(operation_type, operands, root, *, layout, ntt_plaintext=False): ...

Bind compact plaintext indexing and arithmetic to the ciphertext extent.

arithmetic_value ​

function View source

python
def arithmetic_value(operation_type: type[Operation], operands: list[ValueExpr], root: int, *, ntt_plaintext: bool=False) -> ValueExpr: ...

Emitter ​

class View source

python
Emitter(layouts: tuple[InputLayout, ...], lines: list[str] = field(default_factory=list), cache: dict[tuple[Expr, str, str, str, str], str] = field(default_factory=dict))

Emit an expression at arbitrary coefficient, prime-row and batch indices.

Attributes ​

NameTypeDefault/value
layoutstuple[InputLayout, ...]
lineslist[str]field(default_factory=list)
cachedict[tuple[Expr, str, str, str, str], str]field(default_factory=dict)

emit ​

method

python
def emit(expr: Expr, index: str, row: str, batch: str, overrides: Mapping[Expr, str] | None=None, mask: str='mask') -> str: ...