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);
// ...
}
Parent topic: TilingContext