GetWindowsInAddr

Applicability

Product	Supported
Atlas 350 Accelerator Card	x
Atlas A3 training product/Atlas A3 inference product	√
Atlas A2 training product/Atlas A2 inference product	√
Atlas 200I/500 A2 inference product	x
Atlas inference product AI Core	x
Atlas inference product Vector Core	x
Atlas training product	x

Function Usage

Obtains the WindowsIn start address of inter-card communication data, which can be directly used as the input and output address for computation, reducing copy operations. By default, this API works on all cores. You can also specify a core by setting GetBlockIdx before calling this API.

Prototype

__aicore__ inline GM_ADDR GetWindowsInAddr(uint32_t rankId)

Parameters

**Table 1** API parameters
Parameter	Input/Output	Description
rankId	Input	ID of the rank to be queried.

Returns

Returns the WindowsIn start address of inter-card communication data of the corresponding rank. If rankId is invalid, nullptr is returned.

Constraints

None

Examples

REGISTER_TILING_DEFAULT(ReduceScatterCustomTilingData); // ReduceScatterCustomTilingData is a structure defined in the operator header file.
GET_TILING_DATA_WITH_STRUCT(ReduceScatterCustomTilingData, tilingData, tilingGM);
Hccl hccl;
GM_ADDR contextGM = AscendC::GetHcclContext<0>();  // Obtain the HCCL context on the kernel of the Ascend C custom operator.
hccl.InitV2(contextGM, &tilingData);

auto winInAddr = hccl.GetWindowsInAddr(0);
auto winOutAddr = hccl.GetWindowsOutAddr(0);
auto rankId = hccl.GetRankId();
auto rankDim = hccl.GetRankDim();  // Four cards

Parent topic: HCCL Kernel APIs