cuquantum.cutensornet.workspace_get_size

cuquantum.cutensornet.workspace_get_size(intptr_t handle, intptr_t workspace, int pref, int mem_space) uint64_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.

Returns

The computed workspace size.

Return type

uint64_t