Skip to content
voiage

voiage.core.gpu_acceleration.array_to_gpu

array_to_gpu

Section titled “array_to_gpu”
array_to_gpu([positional or keyword] arr: np.ndarray = None, [positional or keyword] backend: str | None = None) -> object

Transfer a NumPy array to GPU memory using the specified or default backend.

Args: arr: NumPy array to transfer to GPU backend: GPU backend to use (‘jax’, ‘cupy’, ‘torch’, or None for auto-detect)

Returns

Section titled “Returns”

Array on GPU (type depends on backend)

Raises

Section titled “Raises”

RuntimeError: If no GPU backend is available

Parameters:

  • arr np.ndarray
  • backend str | None (default: None)

Returns: object