FreeTensor
功能说明
释放Que中的指定Tensor,供Que后续使用。
定义原型
template <typename T> __aicore__ inline void FreeTensor(LocalTensor<T>& tensor);
参数说明
参数名称 |
输入/输出 |
含义 |
---|---|---|
tensor |
输入 |
待释放的Tensor |
支持的型号
Atlas 训练系列产品
Atlas推理系列产品(Ascend 310P处理器)AI Core
Atlas推理系列产品(Ascend 310P处理器)Vector Core
Atlas A2训练系列产品/Atlas 800I A2推理产品
Atlas 200/500 A2推理产品
注意事项
无
返回值
无
调用示例
// 使用FreeTensor释放通过AllocTensor分配的Tensor,注意配对使用 TPipe pipe; TQue<TPosition::VECOUT, 2> que; int num = 4; int len = 1024; pipe.InitBuffer(que, num, len); LocalTensor<half> tensor1 = que.AllocTensor(); que.FreeTensor<half>(tensor1);
父主题: TQue