fhelium.benchmarks.v1.ntt
Fixed cross-backend indexed radix-2 NTT workload for Benchmark v1.
BACKEND
constant View source
python
BACKEND = 'radix2_indexed'1
DEFINITION
constant View source
python
DEFINITION = register_benchmark(BenchmarkDefinition(name=WORKLOAD_ID, title='Indexed radix-2 NTT operations', category='local device', description='Measures one fixed indexed radix-2 NTT workload identically on CPU and CUDA.', profiles=(BenchmarkProfile('core', 'Complete fixed cross-backend NTT cells.', _PARAMETERS),), runner=cast(Any, _run_indexed_ntt), workload_id=WORKLOAD_ID))1
ENTRY_LEVELS
constant View source
python
ENTRY_LEVELS = tuple(range(_CONFIG.num_scale_primes))1
MODULUS_BASES
constant View source
python
MODULUS_BASES: tuple[ModulusBasis, ...] = ('Q', 'QP')1
OPERATIONS
constant View source
python
OPERATIONS: tuple[NttOperation, ...] = ('forward_ntt', 'inverse_ntt', 'roundtrip')1
PRESET
constant View source
python
PRESET = fh.Preset.slots8192_scale40_levels7_int641
WORKLOAD_ID
constant View source
python
WORKLOAD_ID = 'indexed-ntt-operations'1
describe_basis_cell
function View source
python
def describe_basis_cell(config: CkksConfig, *, entry_level: int, modulus_basis: ModulusBasis) -> dict[str, Any]: ...1
make_standard_residue_rows
function View source
python
def make_standard_residue_rows(config: CkksConfig, *, seed: int=SEED) -> torch.Tensor: ...1