fhelium.engine.ntt.backends.compact_radix2
Compact-table grouped radix-2 CUDA NTT production backend.
CompactRadix2NttBackend
class View source
python
CompactRadix2NttBackend(*, policy: CompactRadix2Policy, ntt_tables: CompactRadix2Tables, rns_params: torch.Tensor)1
Compute butterfly indices in CUDA from canonical compact twiddles.
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