昇腾社区首页
中文
注册
开发者
下载

SetDynUBufSize

函数功能

用于设置需要使用的Dynamic Unified Buffer大小。不设置的情况下,默认为0,即算子不需要使用Dynamic Unified Buffer。

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

函数原型

ge::graphStatus SetDynUBufSize(const uint32_t dyn_ubuf_size)

参数说明

参数

输入/输出

说明

local_memory_size

输入

Dynamic Unified Buffer大小。

返回值说明

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

关于graphStatus的定义,请参见ge::graphStatus

约束说明

无。

调用示例

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