SetLocalMemorySize

函数功能

用于设置需要使用的Local Memory大小。不设置的情况下，默认为0，即算子不需要使用Local Memory。

该接口为预留接口，为后续功能做保留，不建议开发者使用，开发者无需关注。

函数原型

ge::graphStatus SetLocalMemorySize(const uint32_t local_memory_size)

参数说明

参数	输入/输出	说明
local_memory_size	输入	Local Memory大小。

返回值说明

设置成功时返回“ge::GRAPH_SUCCESS”。

关于graphStatus的定义，请参见ge::graphStatus。

约束说明

无。

调用示例

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