GetWorkspaceSizes

Description

Obtains the workspace sizes pointer.

Prototype

size_t *GetWorkspaceSizes(const size_t workspace_count)

Parameters

Returns

Pointer to workspace sizes

Restrictions

The number of input workspaces cannot exceed the maximum number of workspaces specified during compilation.

Example

ge::graphStatus Tiling4XXX(TilingContext* context) {
  auto ws = context->GetWorkspaceSizes(5);
  // ...
}