workspace_get_memory_size#

cuquantum.bindings.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