InitBuffer
功能说明
调用TBufPool::InitBuffer接口为TQue/TBuf进行内存分配。
函数原型
1 2 | template <class T> __aicore__ inline bool InitBuffer(T& que, uint8_t num, uint32_t len); template <TPosition pos> __aicore__ inline bool InitBuffer(TBuf<pos>& buf, uint32_t len); |
参数说明
参数名称 |
输入/输出 |
含义 |
|---|---|---|
que |
输入 |
需要分配内存的TQue对象 |
num |
输入 |
分配内存块的个数 |
len |
输入 |
每个内存块的大小，单位为Bytes，非32Bytes对齐会自动向上补齐至32Bytes对齐 |
参数名称 |
输入/输出 |
含义 |
|---|---|---|
buf |
输入 |
需要分配内存的TBuf对象 |
len |
输入 |
为TBuf分配的内存大小，单位为Bytes，非32Bytes对齐会自动向上补齐至32Bytes对齐 |
支持的型号
Atlas 训练系列产品
Atlas推理系列产品AI Core
Atlas A2训练系列产品/Atlas 800I A2推理产品
注意事项
声明TBufPool时，可以通过bufIDSize指定可分配Buffer的最大数量，默认上限为4，最大为16。TQue或TBuf的物理内存需要和TBufPool一致。
返回值
无
调用示例
父主题： TBufPool