华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
__aicore__ inline void Compute(int32_t progress) { LocalTensor<DTYPE_X> inLocal = inQueueIN.DeQue<DTYPE_X>(); LocalTensor<DTYPE_X> xLocal = inLocal; LocalTensor<DTYPE_X> tempTensor1 = calcBuf1.Get<DTYPE_X>(this->tileLength);//tileLength LocalTensor<DTYPE_X> tempTensor2 = calcBuf2.Get<DTYPE_X>();//finalWorkLocalNeedSize LocalTensor<DTYPE_Y> calcLocal = calcQueue.AllocTensor<DTYPE_Y>();//tilenum*BUFFER_NUM if(this->p == 2) { Abs(tempTensor1, xLocal, this->tileLength);
pipe.InitBuffer(inQueueIN, BUFFER_NUM, this->tileLength * sizeof(DTYPE_X)); pipe.InitBuffer(outQueueOUT, BUFFER_NUM, this->tileLength * sizeof(DTYPE_Y)); pipe.InitBuffer(calcQueue,BUFFER_NUM, this->tileNum * BUFFER_NUM* sizeof(DTYPE_Y)); pipe.InitBuffer(calcBuf1,BUFFER_NUM, this->tileLength * sizeof(DTYPE_X));//this->totalLength pipe.InitBuffer(calcBuf2,BUFFER_NUM, this->finalWorkLocalNeedSize * sizeof(DTYPE_X));
pipe.InitBuffer(inQueueIN, BUFFER_NUM, this->tileLength * sizeof(DTYPE_X)); pipe.InitBuffer(outQueueOUT, BUFFER_NUM, this->tileLength * sizeof(DTYPE_Y)); pipe.InitBuffer(calcBuf1,BUFFER_NUM, this->tileLength * sizeof(DTYPE_X));//this->totalLength pipe.InitBuffer(calcQueue,BUFFER_NUM, this->tileNum * BUFFER_NUM* sizeof(DTYPE_Y)); pipe.InitBuffer(calcBuf2,BUFFER_NUM, this->finalWorkLocalNeedSize * sizeof(DTYPE_X));
我要发帖子
已解决
点赞
收藏
回复
分享
举报
__aicore__ inline void Compute(int32_t progress) { LocalTensor<DTYPE_X> inLocal = inQueueIN.DeQue<DTYPE_X>(); LocalTensor<DTYPE_X> xLocal = inLocal; LocalTensor<DTYPE_X> tempTensor1 = calcBuf1.Get<DTYPE_X>(this->tileLength);//tileLength LocalTensor<DTYPE_X> tempTensor2 = calcBuf2.Get<DTYPE_X>();//finalWorkLocalNeedSize LocalTensor<DTYPE_Y> calcLocal = calcQueue.AllocTensor<DTYPE_Y>();//tilenum*BUFFER_NUM if(this->p == 2) { Abs(tempTensor1, xLocal, this->tileLength);pipe.InitBuffer(inQueueIN, BUFFER_NUM, this->tileLength * sizeof(DTYPE_X)); pipe.InitBuffer(outQueueOUT, BUFFER_NUM, this->tileLength * sizeof(DTYPE_Y)); pipe.InitBuffer(calcQueue,BUFFER_NUM, this->tileNum * BUFFER_NUM* sizeof(DTYPE_Y)); pipe.InitBuffer(calcBuf1,BUFFER_NUM, this->tileLength * sizeof(DTYPE_X));//this->totalLength pipe.InitBuffer(calcBuf2,BUFFER_NUM, this->finalWorkLocalNeedSize * sizeof(DTYPE_X));pipe.InitBuffer(inQueueIN, BUFFER_NUM, this->tileLength * sizeof(DTYPE_X)); pipe.InitBuffer(outQueueOUT, BUFFER_NUM, this->tileLength * sizeof(DTYPE_Y)); pipe.InitBuffer(calcBuf1,BUFFER_NUM, this->tileLength * sizeof(DTYPE_X));//this->totalLength pipe.InitBuffer(calcQueue,BUFFER_NUM, this->tileNum * BUFFER_NUM* sizeof(DTYPE_Y)); pipe.InitBuffer(calcBuf2,BUFFER_NUM, this->finalWorkLocalNeedSize * sizeof(DTYPE_X));