CANN/asc-devkit临时空间管理API
PopStackBuffer产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品 / Atlas A3 推理系列产品√Atlas A2 训练系列产品 / Atlas A2 推理系列产品√Atlas 200I/500 A2 推理产品√Atlas 推理系列产品 AI Core√Atlas 推理系列产品 Vector Core√Atlas 训练系列产品√Kirin X90√Kirin 9030√功能说明在指定position逻辑位置申请临时空间空间大小为指定position的全部剩余空间。函数原型template typename T, TPosition pos __aicore__ inline bool PopStackBuffer(LocalTensorT popLocal)参数说明表 1模板参数说明参数名描述TpopLocal的数据类型支持的数据类型如下uint8_t、int8_t、int16_t、uint16_t、int32_t、uint32_t、int64_t、uint64_t、float、half。pos需要申请临时空间的position数据类型为TPosition。表 2参数说明参数名输入/输出描述popLocal输出申请临时空间对应的TensorTensor大小为对应position的剩余全部空间。返回值说明表示函数执行的结果true表示成功false表示失败。约束说明该接口不支持嵌套使用比如函数A中调用了PopStackBuffer那么调用函数A的其他函数中则不可以再调用PopStackBuffer。因为当前高阶API内部实现中会使用到本接口所以算子实现中调用了高阶API的场景不支持调用该接口。调用示例AscendC::LocalTensorint16_t popBuffer; bool ret AscendC::PopStackBufferint16_t, AscendC::TPosition::VECCALC(popBuffer);创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考