rapids_singlecell.get.anndata_to_GPU

rapids_singlecell.get.anndata_to_GPU(adata, layer=None, convert_all=False, copy=False)

Transfers matrices and arrays to the GPU

Parameters:

adata AnnData

AnnData object

layer str | None (default: None)

Layer to use as input instead of X. If None, X is used.

convert_all bool (default: False)

If True, move all supported arrays and matrices on the GPU

copy bool (default: False)

Whether to return a copy or update adata.

Return type:

AnnData | None

Returns:

Updates adata inplace or returns an updated copy