设置通算融合算子每个通信域对应的context（消息区）地址。
template <uint32_t index> __aicore__ inline void SetHcclContext(__gm__ uint8_t* context)
参数名 |
描述 |
|---|---|
index |
模板参数，用来表示要设置的通信域ID，当前只支持2个通信域，index只能为0/1。 |
context |
对应通信域的context（消息区）地址。 |
无
Atlas推理系列产品AI Core
Atlas推理系列产品Vector Core
Atlas A2训练系列产品/Atlas 800I A2推理产品
当前最多只支持2个通信域。
// 给GROUP_0设置消息区地址 AscendC::SetHcclContext<HCCL_GROUP_ID_0>(contextGM);