estimate_workspace_size#

nvmath.bindings.cutensor.estimate_workspace_size( intptr_t handle, intptr_t desc, intptr_t plan_pref, int workspace_pref, ) → uint64_t[source]#

Determines the required workspaceSize for the given operation encoded by desc.

Parameters:

handle (intptr_t) – Opaque handle holding cuTENSOR’s library context.
desc (intptr_t) – This opaque struct encodes the operation.
plan_pref (intptr_t) – This opaque struct restricts the space of viable candidates.
workspace_pref (int) – This parameter influences the size of the workspace; see cutensorWorksizePreference_t for details.

Returns:

The workspace size (in bytes) that is required for the given operation.

Return type:

uint64_t