vibespatial.spatial.segment_primitives_kernels

NVRTC kernel sources for segment extraction, classification, and candidate scatter.

Attributes

Functions

format_extract_source(→ str)

Format the segment extraction kernel source with the given compute type.

format_classify_source(→ str)

Format the classify kernel source with the given compute type and errbound.

Module Contents

vibespatial.spatial.segment_primitives_kernels.format_extract_source(compute_type: str = 'double') str

Format the segment extraction kernel source with the given compute type.

vibespatial.spatial.segment_primitives_kernels.format_classify_source(compute_type: str = 'double') str

Format the classify kernel source with the given compute type and errbound.

vibespatial.spatial.segment_primitives_kernels.EXTRACT_SOURCE_FP64
vibespatial.spatial.segment_primitives_kernels.EXTRACT_SOURCE_FP32
vibespatial.spatial.segment_primitives_kernels.CLASSIFY_SOURCE_FP64
vibespatial.spatial.segment_primitives_kernels.CLASSIFY_SOURCE_FP32