FreeTensor

Function Usage

Releases a specified tensor in a queue for subsequent use.

Prototype

template <typename T>
__aicore__ inline void FreeTensor(LocalTensor<T>& tensor)

Parameters

Parameter	Input/Output	Meaning
tensor	Input	Tensor to be released

Availability

Atlas Training Series Product

Precautions

None

Returns

None

Example

// Use FreeTensor to release the tensors allocated by AllocTensor. Note that FreeTensor needs to be used together with AllocTensor.
AscendC::TPipe pipe;
AscendC::TQue<AscendC::TPosition::VECOUT, 2> que;
int num = 4;
int len = 1024;
pipe.InitBuffer(que, num, len);
AscendC::LocalTensor<half> tensor1 = que.AllocTensor<half>();
que.FreeTensor<half>(tensor1);

Parent topic: TQue