FHElium

Appearance

fhelium.backend.ntt.plans.compact_radix2 ​

Compact radix-2 NTT plan construction.

CompactRadix2NttPlan ​

class View source

python
CompactRadix2NttPlan(ckks_config: CkksConfig, *, device: str | int | torch.device | None=None, dtype: torch.dtype | None=None)

Compact forward and inverse standard-residue twiddle rows.

Compact CUDA kernels compute butterfly indices in-kernel and consume one integral [prime, coefficient] twiddle table per transform direction. Prime rows follow ckks_config.moduli exactly; final extent is N and both tables use the selected execution dtype on device. Construction is functional and the forward and inverse tables do not alias.