SetLocalMemorySize

Function Usage

Sets the size of the dynamic unified buffer to be used for SIMT operators. If this parameter is not set, the default value 0 is used, indicating that the operator does not need to use the dynamic unified buffer.

This API has been deprecated and will be removed in later versions. Use SetDynUBufSize to set the required size of the dynamic unified buffer.

Prototype

ge::graphStatus SetLocalMemorySize(const uint32_t local_memory_size)

Parameters

Returns

ge::GRAPH_SUCCESS on success.

For details about the definition of graphStatus, see ge::graphStatus.

Constraints

None

Examples

ge::graphStatus Tiling4XXX(TilingContext* context) {
    // ...
    auto ret = context->SetLocalMemorySize(1024 * 128);
}