FreeTensor

Supported Products

Product

Supported/Unsupported

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 Usage

Frees a specified tensor in the queue.

Prototype

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

Parameters

Table 1 Parameters in the template

Parameter

Description

T

Tensor data type,

Table 2 Parameters

Parameter

Input/Output

Meaning

tensor

Input

Tensor to be released.

Restrictions

None

Return Value Description

None

Example

1
2
3
4
5
6
7
8
// 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);