GetWorkspaceSizes

Function Usage

Obtains the point to workspace sizes (in bytes).

Prototype

size_t *GetWorkspaceSizes(const size_t workspace_count)

Parameters

Parameter	Input/Output	Description
workspace_count	Input	Number of workspaces. The value cannot exceed the number of workspaces returned by GetWorkspaceNum. If the value exceeds the number of workspaces, an empty pointer is returned.

Returns

Pointer to workspace sizes

Constraints

None

Examples

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

Parent topic: TilingContext