SetCommBlockNum

Function

Sets the number of cores involved in BatchWrite communication.

Prototype

1
uint32_t SetCommBlockNum(uint16_t num)

Parameters

Table 1 Parameters

Parameter

Input/Output

Description

num

Input

Number of cores.

Returns

  • 0: Setting succeeded.
  • Other values: Setting failed.

Restrictions

This API takes effect only when the communication type is HCCL_CMD_BATCH_WRITE on the Atlas A3 training products/Atlas A3 inference products.

Example

1
2
3
4
5
6
const char *groupName = "testGroup";
uint32_t opType = HCCL_CMD_BATCH_WRITE;
std::string algConfig = "BatchWrite=level0:fullmesh";
uint32_t reduceType = HCCL_REDUCE_SUM;
AscendC::Mc2CcTilingConfig mc2CcTilingConfig(groupName, opType, algConfig, reduceType);
mc2CcTilingConfig.SetCommBlockNum(24U);