SetHcclContext

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

Atlas inference product Vector Core

Atlas training product

x

Function Usage

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 a communicator.

Returns

None

Restrictions

Currently, a maximum of two communicators are supported.

Examples

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