SetHcclContext

Applicability

Product

Supported

Atlas A3 training products/Atlas A3 inference products

Atlas A2 training products/Atlas A2 inference products

Atlas 200I/500 A2 inference products

x

Atlas inference product's AI Core

Atlas inference product's Vector Core

Atlas training products

x

Function

Sets the context (message area) address of each communicator of an MC2 operator.

Prototype

1
2
template <uint32_t index>
__aicore__ inline void SetHcclContext(__gm__ uint8_t* context)

Parameters

Table 1 Parameters

Parameter

Description

index

A template parameter, which indicates the communicator ID to be set. Currently, only two communicators are supported. index can be 0 or 1.

context

Context (message area) address of the corresponding communicator.

Returns

None

Restrictions

Currently, a maximum of two communicators are supported.

Example

1
2
// Set the message area address for GROUP_0.
AscendC::SetHcclContext<0>(contextGM);