CANN/asc-devkit 弹出栈缓冲区API文档

发布时间:2026/7/17 15:22:15

CANN/asc-devkit 弹出栈缓冲区API文档 PopStackBuffer【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit产品支持情况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);【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

相关新闻