GetWorkspaceSizes
函数功能
获取workspace sizes指针。
函数原型
size_t *GetWorkspaceSizes(const size_t workspace_count);
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
workspace_count |
输入 |
workspace的个数,传入的workspace个数不可以超过编译时指定的最大workspace个数。 |
返回值说明
workspace sizes指针
约束说明
传入的workspace个数不可以超过编译时指定的最大workspace个数。
调用示例
ge::graphStatus Tiling4XXX(TilingContext* context) { auto ws = context->GetWorkspaceSizes(5); // ... }
父主题: TilingContext