fhelium.ir.dialects
Registered dialect modules for FHElium's multi-depth IR.
FHElium
data View source
python
FHElium = Dialect('fhelium', [MaterialRefOp, ResourceRefOp, ConstantOp], [EncryptedType, MessageType, PlaintextType, MaterialType, ResourceType])1
FHEliumCkks
data View source
python
FHEliumCkks = Dialect('fhelium_ckks', [PrepareCompressedPlaintextOp, EncodeOp, DecodeOp, IntegerCoefficientsToRnsOp, EncryptOp, DecryptOp, NegateOp, RotateOp, RotateManyOp, GroupedRotationWeightedSumOp, ToNttOp, FromNttOp, ToMontgomeryResiduesOp, ToStandardResiduesOp, AddOp, SumBatchOp, SubtractOp, MultiplyOp, AddScalarOp, MultiplyScalarOp, MultiplyIntegerScalarOp, AddPlaintextOp, MultiplyPlaintextOp, AddCompressedPlaintextOp, MultiplyCompressedPlaintextOp, RelinearizeOp, SwitchKeyOp, ConjugateOp, RescaleOp, ModSwitchOp, ReinterpretScaleOp, PrepareAddMessageOp, PrepareAddPlaintextOp, PrepareAddStaticOp, PrepareMultiplyMessageOp, PrepareMultiplyPlaintextOp, PrepareMultiplyStaticOp], [CiphertextType, PlaintextType, CompressedPlaintextType, EvaluationKeyType])1
FHEliumDistributed
data View source
python
FHEliumDistributed = Dialect('fhelium_dist', [RankOp, GroupSizeOp, BroadcastOp, AllReduceOp, AllReduceAddCiphertextOp, YieldOp], [GroupType])1
FHEliumFusion
data View source
python
FHEliumFusion = Dialect('fhelium_fusion', [FusedOp, YieldOp], [])1
FHEliumLogical
data View source
python
FHEliumLogical = Dialect('fhelium_logical', [*_LOGICAL_OPERATION_TYPES], [EncryptedType, PublicType])1
FHEliumMemory
data View source
python
FHEliumMemory = Dialect('fhelium_memory', [TransferOp], [DeviceType])1
FHEliumNtt
data View source
python
FHEliumNtt = Dialect('fhelium_ntt', [*_NTT_OPERATION_TYPES], [])1
FHEliumRns
data View source
python
FHEliumRns = Dialect('fhelium_rns', [*_RNS_OPERATION_TYPES], [RnsBundleType, RnsParametersType])1
FHEliumSemantic
data View source
python
FHEliumSemantic = Dialect('fhelium_semantic', [AddOp, SubtractOp, MultiplyOp, NegateOp, RollOp], [SecretType, PublicType])1
REGISTERED_DIALECTS
constant View source
python
REGISTERED_DIALECTS = (FHElium, FHEliumSemantic, FHEliumLogical, FHEliumCkks, FHEliumRns, FHEliumNtt, FHEliumMemory, FHEliumDistributed, FHEliumFusion, Torch)1
Torch
data View source
python
Torch = Dialect('torch', [CallOp, TensorCallOp], [])1
ckks
module
fhelium.ir.dialects.ckks
core
module
fhelium.ir.dialects.core
distributed
module
fhelium.ir.dialects.distributed
fusion
module
fhelium.ir.dialects.fusion
logical
module
fhelium.ir.dialects.logical
memory
module
fhelium.ir.dialects.memory
ntt
module
fhelium.ir.dialects.ntt
rns
module
fhelium.ir.dialects.rns
semantic
module
fhelium.ir.dialects.semantic
torch
module
fhelium.ir.dialects.torch