asc.language.basic.mrg_sort【免费下载链接】pyasc本项目为Python用户提供算子编程接口支持在昇腾AI处理器上加速计算接口与Ascend C一一对应并遵守Python原生语法。项目地址: https://gitcode.com/cann/pyascasc.language.basic.mrg_sort(dst: LocalTensor, sort_list: MrgSortSrcList, element_count_list: List[int], sorted_num: List[int], valid_bit: int, repeat_time: int, is_exhausted_suspension: bool False) → Noneasc.language.basic.mrg_sort(dst: LocalTensor, sort_list: MrgSortSrcList, params: MrgSort4Info) → None将已经排好序的多个队列合并成一条队列结果按照指定顺序排序。对应的Ascend C函数原型template typename T __aicore__ inline void MrgSort(const LocalTensorT dst, const MrgSortSrcListT src, const uint16_t elementCountList[], uint32_t sortedNum[], uint16_t validBit, uint16_t repeatTime, bool isExhaustedSuspension false)template typename T __aicore__ inline void MrgSort(const LocalTensorT dst, const MrgSortSrcListT src, const MrgSort4Info params)参数说明dst (asc.LocalTensor): 目的操作数存储经过排序后的结果。支持的TPosition为VECIN/VECCALC/VECOUT。src (asc.MrgSortSrcList): 源操作数多个已经排好序的队列。具体定义如下class MrgSortSrcList: src1: LocalTensor # 第一个已经排好序的Region Proposals队列 src2: LocalTensor # 第二个已经排好序的Region Proposals队列 src3: LocalTensor # 第三个已经排好序的Region Proposals队列 src4: LocalTensor # 第四个已经排好序的Region Proposals队列element_count_list: 各个源队列的长度元素数目类型为uint16_t数组。sorted_num: 输出参数存储各个队列排序后的元素数目类型为uint32_t数组。valid_bit: 有效队列个数位掩码。repeat_time: 迭代次数。is_exhausted_suspension: 某条队列耗尽后指令是否需要停止类型为bool默认false。params (asc.MrgSort4Info)排序所需参数。element_lengths: 四个源Region Proposals队列的长度Region Proposal数目类型为长度为4的uint16_t数组每个元素取值范围[0, 4095]。is_exhausted_suspension: 某条队列耗尽后指令是否需要停止类型为bool默认false。valid_bit有效队列个数。repeat_times迭代次数每一次源操作数和目的操作数跳过四个队列总长度。取值范围[1,255]。约束说明操作数地址对齐要求请参见 《Ascend C算子开发接口》 中的“通用说明和约束-通用地址对齐约束”。不支持源操作数与目的操作数之间存在地址重叠。调用示例src_list asc.MrgSortSrcList(queue1, queue2, queue3, queue4) element_counts [16, 16, 16, 16] sorted_nums [0, 0, 0, 0] valid_bit 15 # 所有4个队列都有效 repeat_time 1 asc.mrg_sort(dst, src_list, element_counts, sorted_nums, valid_bit, repeat_time) src_list asc.MrgSortSrcList(queue1, queue2, queue3, queue4) element_lengths [16, 16, 16, 16] params asc.MrgSort4Info(element_lengths, False, 15, 1) asc.mrg_sort(dst, src_list, params)【免费下载链接】pyasc本项目为Python用户提供算子编程接口支持在昇腾AI处理器上加速计算接口与Ascend C一一对应并遵守Python原生语法。项目地址: https://gitcode.com/cann/pyasc创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考