GetTensorCountInQue

Product Support

Product	Supported
Atlas A3 training products / Atlas A3 inference products	√
Atlas A2 training products / Atlas A2 inference products	√
Atlas 200I/500 A2 inference products	√
Atlas inference product 's AI Core	√
Atlas inference product 's Vector Core	x
Atlas training products	√

Function

Queries the number of enqueued tensors in a queue.

Prototype

      
           __aicore__ inline int32_t GetTensorCountInQue()

Parameters

None

Restrictions

This API does not support inplace tensor operations, that is, the scenario where the depth of TQue is set to 0.

Returns

Number of enqueued tensors in the queue

Example

      
           // Call GetTensorCountInQue to query the number of enqueued tensors in the queue. Currently, AllocTensor is called to allocate memory and add the memory to the queue. The value of num is 1.
AscendC::TPipe pipe;
AscendC::TQue<AscendC::TPosition::VECOUT, 4> que;
int num = 4;
int len = 1024;
pipe.InitBuffer(que, num, len);
AscendC::LocalTensor<half> tensor1 = que.AllocTensor<half>();
que.EnQue(tensor1);// Add the tensor to the queue of VECOUT.
int32_t numb = que.GetTensorCountInQue();

Parent topic: TQue