获取卡间通信数据WindowsOut起始地址,可用来直接作为计算的输入输出地址,减少拷贝。该接口默认在所有核上工作,用户也可以在调用前通过GetBlockIdx指定其在某一个核上运行。
1 | __aicore__ inline GM_ADDR GetWindowsOutAddr(uint32_t rankId) |
参数名 |
输入/输出 |
描述 |
---|---|---|
rankId |
输入 |
待查询的卡的Id。 |
返回对应卡的卡间通信数据WindowsOut起始地址。当rankId非法时,返回nullptr。
Atlas A2训练系列产品/Atlas 800I A2推理产品
该接口只能在AI Cube核或者AI Vector核两者之一上调用
请参见GetWindowsInAddr的调用示例。