warp.utils

Utilities supporting Warp’s high-level workflows.

Array Operations

array_cast	Cast elements from one array to another array with a different data type.
array_inner	Compute the inner product of two arrays.
array_scan	Perform a scan (prefix sum) operation on an array.
array_sum	Compute the sum of array elements.

Sorting

radix_sort_pairs	Sort key-value pairs using radix sort.
runlength_encode	Perform run-length encoding on an array.
segmented_sort_pairs	Sort key-value pairs within segments.

Graph Coloring

GraphColoringAlgorithm	Graph coloring algorithm selection.
graph_coloring_assign	Assign colors to graph nodes such that no two adjacent nodes share the same color.
graph_coloring_balance	Balance the sizes of color groups in a graph coloring.
graph_coloring_get_groups	Convert node colors into per-color groups.

Misc

create_warp_function

Create a Warp function from a Python function.