CANN/asc-devkit VacantInQue函数
VacantInQue
产品支持情况
功能说明
查询队列是否已满。
函数原型
__aicore__ inline bool VacantInQue()参数说明
无
约束说明
该接口不支持Tensor原地操作,即TQue的depth设置为0的场景。
返回值说明
- true - 表示Queue未满,可以继续Enque操作
- false - 表示Queue已满,不可以继续入队
调用示例
// 根据VacantInQue判断当前que是否已满,设置当前队列深度为4 AscendC::TPipe pipe; AscendC::TQue<AscendC::TPosition::VECOUT, 4> que; int num = 10; int len = 1024; pipe.InitBuffer(que, num, len); bool ret = que.VacantInQue(); // 返回为true AscendC::LocalTensor<half> tensor1 = que.AllocTensor<half>(); AscendC::LocalTensor<half> tensor2 = que.AllocTensor<half>(); AscendC::LocalTensor<half> tensor3 = que.AllocTensor<half>(); AscendC::LocalTensor<half> tensor4 = que.AllocTensor<half>(); AscendC::LocalTensor<half> tensor5 = que.AllocTensor<half>(); que.EnQue(tensor1);// 将tensor1加入VECOUT的Queue中 que.EnQue(tensor2);// 将tensor2加入VECOUT的Queue中 que.EnQue(tensor3);// 将tensor3加入VECOUT的Queue中 que.EnQue(tensor4);// 将tensor4加入VECOUT的Queue中 ret = que.VacantInQue(); // 返回为false, 继续入队操作(Enque)将报错创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考