fhelium.engine.ntt.backends.power_of_two_radix
Compact-table power-of-two-radix CUDA NTT backend.
CompactPowerOfTwoRadixNttBackend
class View source
python
CompactPowerOfTwoRadixNttBackend(*, policy: CompactFixedRadixPolicy, ntt_tables: CompactPowerOfTwoRadixTables, rns_params: torch.Tensor)1
Execute radix-4/8/16 digits with radix-specific CUDA butterflies.
forward_montgomery_
method
python
def forward_montgomery_(operand: torch.Tensor, parameter_row_start: int) -> None: ...1
forward_to_montgomery_
method
python
def forward_to_montgomery_(operand: torch.Tensor, parameter_row_start: int) -> None: ...1
forward_to_montgomery
method
python
def forward_to_montgomery(operand: torch.Tensor, parameter_row_start: int) -> torch.Tensor: ...1
inverse_montgomery_
method
python
def inverse_montgomery_(operand: torch.Tensor, parameter_row_start: int) -> None: ...1
inverse_to_standard_lazy_
method
python
def inverse_to_standard_lazy_(operand: torch.Tensor, parameter_row_start: int) -> None: ...1
inverse_to_standard_
method
python
def inverse_to_standard_(operand: torch.Tensor, parameter_row_start: int) -> None: ...1
inverse_to_centered_
method
python
def inverse_to_centered_(operand: torch.Tensor, parameter_row_start: int) -> None: ...1