文档
注册

SetBlockDim

函数功能

设置block dim,即参与计算的VectorCore或者CubeCore核数。

函数原型

ge::graphStatus SetBlockDim(const uint32_t block_dim)

参数说明

参数

输入/输出

说明

block_dim

输入

参与计算的VectorCore或者CubeCore核数。注意这里是逻辑核的概念,取值范围为[1,65535]。为了充分利用硬件资源,一般设置为物理核的核数或其倍数。特别地,对于mix场景(同时用到CubeCore和VectorCore),通常以较小的核数(一般为CubeCore)作为参考标准。VectorCore的核数和CubeCore的核数分别通过GetCoreNumAivGetCoreNumAic接口获取。

返回值说明

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

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

约束说明

调用示例

ge::graphStatus Tiling4XXX(TilingContext* context) {
  auto ret = context->SetBlockDim(32);
  // ...
}
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词