SetAicpuBlockDim

Function Usage

Sets the number of AI CPU cores that can be scheduled.

When using the HCCL high-level API BatchWrite, you can set the number of AI CPU cores appropriately to obtain better performance.

Prototype

ge::graphStatus SetAicpuBlockDim(uint32_t block_dim)

Parameters

Parameter

Input/Output

Description

block_dim

Input

Number of AI CPU cores that can be scheduled.

Returns

ge::GRAPH_SUCCESS on success.

For details about the definition of graphStatus, see ge::graphStatus.

Constraints

None

Examples

ge::graphStatus Tiling4XXX(TilingContext* context) {
  auto ret = context->SetAicpuBlockDim(5U);
  // ...
}