跳到主要内容

GetWorkspaceSizes

函数功能

获取workspace sizes指针。

函数原型

size_t *GetWorkspaceSizes(const size_t workspace_count);

参数说明

参数输入/输出说明
workspace_count输入workspace的个数,传入的workspace个数不可以超过编译时指定的最大workspace个数。

返回值

workspace sizes指针。

约束说明

传入的workspace个数不可以超过编译时指定的最大workspace个数。

当前Kirin9020支持的最大的workspace是8个。

当前KirinX90支持的最大的workspace是8个。

调用示例

ge::graphStatus Tiling4XXX(TilingContext* context) {
auto ws = context->GetWorkspaceSizes(5);
// ...
}