Skip to main content

download_tensor

Function download_tensor 

Source
pub fn download_tensor(rt: &CudaRuntime, tensor: &Tensor) -> Result<Tensor>
Expand description

Download a CubeCL-managed GPU tensor back to host memory.

§Examples

use tenferro_gpu::{cuda::download_tensor, cuda::CudaRuntime};
use tenferro_tensor::{Result, Tensor};

let _download: fn(&CudaRuntime, &Tensor) -> Result<Tensor> = download_tensor;

§Errors

Returns crate::Error::RuntimeState for a host-backed or foreign tensor, crate::Error::BackendSource when synchronization/readback fails, or a typed validation error when device data cannot be decoded.