warp.utils#
Utility functions for array operations, sorting, and function creation.
This module provides GPU-accelerated utility functions for common array operations (scans, reductions, transformations), sorting algorithms (radix sort, segmented sort, run-length encoding), and a utility for creating Warp functions from Python callables.
API#
Cast elements from one array to another array with a different data type. |
|
Compute the inner product of two arrays. |
|
Perform a scan (prefix sum) operation on an array. |
|
Compute the sum of array elements. |
|
Create a Warp function from a Python function. |
|
Sort key-value pairs using radix sort. |
|
Perform run-length encoding on an array. |
|
Sort key-value pairs within segments. |