CANN/asc-devkit Extract排序提取函数

Extract【免费下载链接】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 训练系列产品不支持功能说明处理Sort的结果数据输出排序后的value和index。函数原型template typename T __aicore__ inline void Extract(const LocalTensorT dstValue, const LocalTensoruint32_t dstIndex, const LocalTensorT sorted, const int32_t repeatTime)参数说明表1模板参数说明参数名描述T操作数的数据类型。支持的数据类型为half、float。表2参数说明参数名输入/输出含义dstValue输出目的操作数。类型为LocalTensor支持的TPosition为VECIN/VECCALC/VECOUT。LocalTensor的起始地址需要32字节对齐。dstIndex输出目的操作数。类型为LocalTensor支持的TPosition为VECIN/VECCALC/VECOUT。LocalTensor的起始地址需要32字节对齐。此源操作数固定为uint32_t数据类型。sorted输入源操作数。类型为LocalTensor支持的TPosition为VECIN/VECCALC/VECOUT。LocalTensor的起始地址需要32字节对齐。源操作数的数据类型需要与目的操作数保持一致。repeatTime输入重复迭代次数int32_t类型。Ascend 950PR/Ascend 950DT每次迭代处理64个float类型或128个half类型元素。Atlas A3 训练系列产品/Atlas A3 推理系列产品每次迭代处理64个float类型或128个half类型元素。Atlas A2 训练系列产品/Atlas A2 推理系列产品每次迭代处理64个float类型或128个half类型元素。Atlas 推理系列产品AI Core每次迭代完成16个Region Proposals的元素抽取并排布到16个元素里下次迭代跳至相邻的下一组16个Region Proposals和下一组16个元素。取值范围repeatTime∈[0,255]。返回值说明无约束说明操作数地址对齐要求请参见通用地址对齐约束。调用示例请参见MrgSort的调用示例。【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考