cuquantum.cutensornet.contraction_get_workspace_size¶
- cuquantum.cutensornet.contraction_get_workspace_size(intptr_t handle, intptr_t tn_desc, intptr_t info) uint64_t¶
-
Compute the required workspace size for contracting the input tensor network.
- Parameters
-
handle (intptr_t) – The library handle.
tn_desc (intptr_t) – the tensor network descriptor.
info (intptr_t) – The optimizer info handle.
- Returns
-
The workspace size (in bytes).
- Return type
-
uint64_t
Note
This function should be called either after a contraction path is manually set, or after
contraction_optimize()is called.