开发者
资源

SetAicpuNumBlocks

函数功能

设置可以调度的AI CPU核数。

在使用HCCL高阶API的BatchWrite接口时,可以通过合理设置AI CPU核数获得更好的性能。

函数原型

1
ge::graphStatus SetAicpuNumBlocks(uint32_t num_blocks)

参数说明

参数

输入/输出

说明

SetAicpuNumBlocks

输入

可以调度的AI CPU核数。

返回值说明

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

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

约束说明

调用示例

1
2
3
4
ge::graphStatus Tiling4XXX(TilingContext* context) {
  auto ret = context->SetAicpuNumBlocks(5U);
  // ...
}