cuquantum.cutensornet.workspace_get_memory_size¶

cuquantum.cutensornet.workspace_get_memory_size(intptr_t handle, intptr_t work_desc, int work_pref, int mem_space, int work_kind) → int64_t[source]¶

Retrieves the needed workspace size for the given workspace preference, memory space, workspace kind.

Parameters

handle (intptr_t) – Opaque handle holding cuTensorNet’s library context.
work_desc (intptr_t) – Opaque structure describing the workspace.
work_pref (WorksizePref) – Preference of workspace for planning.
mem_space (Memspace) – The memory space where the workspace is allocated.
work_kind (WorkspaceKind) – The kind of workspace.

Returns

Needed workspace size.

Return type

int64_t