vibespatial.spatial.segment_primitives_kernels¶
NVRTC kernel sources for segment extraction, classification, and candidate scatter.
Attributes¶
Functions¶
|
Format the segment extraction kernel source with the given compute type. |
|
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¶