asc.language.basic.set_hccl_context
- asc.language.basic.set_hccl_context(index: int, context: GlobalAddress) None
设置通算融合算子每个通信域对应的context(消息区)地址。
对应的Ascend C函数原型
template <uint32_t index> __aicore__ inline void SetHcclContext(__gm__ uint8_t* context)
参数说明
index: 模板参数,用来表示要设置的通信域ID,当前只支持2个通信域,index只能为0/1。
context: 对应通信域的context(消息区)地址。
约束说明
当前最多只支持2个通信域。
调用示例
asc.set_hccl_context(0, x)