CuPy-specific functions#
CuPy-specific functions are placed under cupyx
namespace.
Returns the reciprocal square root. |
|
|
Adds given values to specified elements of an array. |
|
Stores a maximum value of elements specified by indices to an array. |
|
Stores a minimum value of elements specified by indices to an array. |
|
Returns a new, uninitialized NumPy array with the given shape and dtype. |
|
Returns a new, uninitialized NumPy array with the same shape and dtype as those of the given array. |
|
Returns a new, zero-initialized NumPy array with the given shape and dtype. |
|
Returns a new, zero-initialized NumPy array with the same shape and dtype as those of the given array. |
Profiling utilities#
|
Timing utility for measuring time spent by both CPU and GPU. |
|
Mark function calls with ranges using NVTX/rocTX. |
Enable CUDA profiling during with statement. |
DLPack utilities#
Below are helper functions for creating a cupy.ndarray
from either a DLPack tensor
or any object supporting the DLPack data exchange protocol.
For further detail see DLPack.
|
Zero-copy conversion between array objects compliant with the DLPack data exchange protocol. |
Automatic Kernel Parameters Optimizations (cupyx.optimizing
)#
|
Context manager that optimizes kernel launch parameters. |