函数原型
size_t *GetWorkspaceSizes(const size_t workspace_count);
参数说明
| 参数 | 输入/输出 | 说明 |
|---|
| workspace_count | 输入 | workspace的个数，传入的workspace个数不可以超过编译时指定的最大workspace个数。 |
约束说明
传入的workspace个数不可以超过编译时指定的最大workspace个数。
当前Kirin9020/Kirin9030/KirinX90支持的最大的workspace是8个。
调用示例
ge::graphStatus Tiling4XXX(TilingContext* context) {
auto ws = context->GetWorkspaceSizes(5);
// ...
}