SetLocalMemorySize

Description

Sets the size of the local memory to be used. If this parameter is not set, the default value 0 is used, indicating that the operator does not need to use the local memory.

This API is reserved for subsequent functions. You are not advised to use this API.

Prototype

ge::graphStatus SetLocalMemorySize(const uint32_t local_memory_size)

Parameters

Parameter

Input/Output

Description

local_memory_size

Input

Size of the local memory.

Returns

ge::GRAPH_SUCCESS on success.

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

Restrictions

None

Example

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