vibespatial.runtime.cccl_warmup_specs

Attributes

cp

Classes

AlgorithmFamily	Enum where members are also (and must be) strings
CCCLWarmupSpec	Specification for a CCCL make_* pre-compilation target.
WarmupDiagnostic

Functions

build_spec_registry(→ dict[str, CCCLWarmupSpec])

Module Contents

vibespatial.runtime.cccl_warmup_specs.cp = None

class vibespatial.runtime.cccl_warmup_specs.AlgorithmFamily

Enum where members are also (and must be) strings

EXCLUSIVE_SCAN = 'exclusive_scan'

SELECT = 'select'

REDUCE_INTO = 'reduce_into'

SEGMENTED_REDUCE = 'segmented_reduce'

LOWER_BOUND = 'lower_bound'

UPPER_BOUND = 'upper_bound'

RADIX_SORT = 'radix_sort'

MERGE_SORT = 'merge_sort'

UNIQUE_BY_KEY = 'unique_by_key'

SEGMENTED_SORT = 'segmented_sort'

class vibespatial.runtime.cccl_warmup_specs.CCCLWarmupSpec

Specification for a CCCL make_* pre-compilation target.

name: str

family: AlgorithmFamily

key_dtype: Any

value_dtype: Any | None

op_name: str

class vibespatial.runtime.cccl_warmup_specs.WarmupDiagnostic

name: str

elapsed_ms: float

success: bool

error: str = ''

vibespatial.runtime.cccl_warmup_specs.build_spec_registry() → dict[str, CCCLWarmupSpec]