fhelium.backend.ntt.automorphism
Galois permutations of bit-reversed NTT evaluations.
ntt_galois_indices
function View source
python
def ntt_galois_indices(n: int, galois_element: int, device: torch.device) -> torch.Tensor: ...1
Map stored bit-reversed NTT positions through the Galois automorphism.
NativeNttAutomorphismImplementation
class View source
python
NativeNttAutomorphismImplementation()1
Execute the supplied NTT index gather without modifying its operands.
Attributes
| Name | Type | Default/value |
|---|---|---|
name | 'native-ntt-automorphism' | |
supports_in_place | False | |
operation_types | tuple[type[Operation], ...] | (AutomorphismOp,) |
resource_requirements
method
python
def resource_requirements(invocation: OperationInvocation) -> tuple[ResourceRequirement, ...]: ...1
execute
method
python
def execute(invocation: OperationInvocation, values: tuple[torch.Tensor, ...], resources: tuple[BoundResource, ...], *, in_place: bool) -> tuple[torch.Tensor, ...]: ...1