EnQue

Function Usage

Pushes a tensor to a queue.

Prototype

1
2
template <typename T>
__aicore__ inline bool EnQue(const LocalTensor<T>& tensor)

Parameters

Table 1 Parameters in bool EnQue(LocalTensor<T>& tensor) prototype definition

Parameter

Input/Output

Meaning

tensor

Input

Specified tensor

Figure 1 Moving the LocalTensor into the queue of A1/B1 through EnQue

Availability

Atlas Training Series Product

Precautions

None

Returns

  • true: The tensor is successfully added to the queue.
  • false: The queue is full and the addition fails.

Example

1
2
3
4
5
6
7
8
// API: EnQue Tensor
AscendC::TPipe pipe;
AscendC::TQueBind<AscendC::TPosition::VECOUT, AscendC::TPosition::GM, 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.