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);
}
Parent topic: TilingContext