CANN/asc-devkit:GetTPipePtr函数
GetTPipePtr
产品支持情况
功能说明
创建TPipe对象时,对象初始化会设置全局唯一的TPipe指针。本接口用于获取该指针,获取该指针后,可进行TPipe相关的操作。
函数原型
__aicore__ inline AscendC::TPipe* GetTPipePtr()约束说明
无
调用示例
如下样例中,在核函数入口处创建TPipe对象,对象初始化会设置全局唯一的TPipe指针。在调用KernelAdd类Init函数时,无需显式传入TPipe指针,而是在函数内直接使用GetTPipePtr获取全局TPipe指针,用来做InitBuffer等操作。如需运行,请参考GetTPipePtr样例实现完整的代码。
// inQueueX、inQueueY为VECIN上的TQue,outQueueZ为VECOUT上的TQue // 获取全局TPipe指针,使用TPipe的InitBuffer接口给TQue分配内存 GetTPipePtr()->InitBuffer(inQueueX, 2, this->tileLength * sizeof(float)); GetTPipePtr()->InitBuffer(inQueueY, 2, this->tileLength * sizeof(float)); GetTPipePtr()->InitBuffer(outQueueZ, 2, this->tileLength * sizeof(float));创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
