跳到主要内容

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);
// ...
}