cuquantum.cutensornet.workspace_get_memory_size¶

cuquantum.cutensornet.workspace_get_memory_size(intptr_t handle, intptr_t workspace, int pref, int mem_space, int kind) → int64_t[source]¶

Get the workspace size for the corresponding preference and memory space. Must be called after workspace_compute_sizes().

Parameters

handle (intptr_t) – The library handle.
workspace (intptr_t) – The workspace descriptor.
pref (WorksizePref) – The preference for the workspace size.
mem_space (Memspace) – The memory space for the workspace being queried.
kind (WorkspaceKind) – The kind of the workspace being queried

Returns

The computed workspace size.

Return type

int64_t