Function cuda_memcpy_h2d

samples/interoperability/cuda_python_common.py:80–104

Copy host array to device array. Args: host_array: Host array to copy. device_array: Device array to copy to, __cuda_array_interface__ or object with interface.

(
    host_array: np.ndarray,
    device_array: int | dict | object,
)

# docs_tag: being_cuda_memcpy_h2d
def cuda_memcpy_h2d(
    host_array: np.ndarray,
    device_array: int | dict | object,
) -> None:
    """
    Copy host array to device array.

    Args:
        host_array: Host array to copy.
        device_array: Device array to copy to, __cuda_array_interface__ or object with interface.
    """
    if hasattr(device_array, "__cuda_array_interface__"):
        device_array = device_array.__cuda_array_interface__["data"][0]
    elif isinstance(device_array, dict) and "data" in device_array:
        device_array = device_array["data"][0]
    elif not isinstance(device_array, int):
        raise ValueError("Invalid device array")
    (err,) = cudart.cudaMemcpy(
        device_array,
        host_array.ctypes.data,
        host_array.nbytes,
        cudart.cudaMemcpyKind.cudaMemcpyHostToDevice,
    )
    if err != cudart.cudaError_t.cudaSuccess:
        raise RuntimeError(f"cudaMemcpy failed: {err}")


# docs_tag: end_cuda_memcpy_h2d