中昊芯英(杭州)科技有限公司闯小明获国家专利权
买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
龙图腾网获悉中昊芯英(杭州)科技有限公司申请的专利用于张量处理单元的张量维度重组方法及装置、芯片获国家发明授权专利权,本发明授权专利权由国家知识产权局授予,授权公告号为:CN121542216B 。
龙图腾网通过国家知识产权局官网在2026-05-12发布的发明授权授权公告中获悉:该发明授权的专利申请号/专利号为:202610084770.1,技术领域涉及:G06F15/80;该发明授权用于张量处理单元的张量维度重组方法及装置、芯片是由闯小明;李海;高育欢;马辉;蒋壹钦;申屠克勤;钱炟才设计研发完成,并于2026-01-22向国家知识产权局提交的专利申请。
本用于张量处理单元的张量维度重组方法及装置、芯片在说明书摘要公布了:本申请涉及张量处理领域,提供了一种用于张量处理单元的张量维度重组方法及装置、芯片。该方法包括:校验输入张量与目标输出张量的元素总数是否相等;基于输入张量和目标输出张量的形状参数、存储布局信息以及张量处理单元的硬件架构特性,对当前维度重组操作进行模式判定,以判断当前维度重组操作是否匹配预设的高频特判模式;当前维度重组操作匹配高频特判模式时,采用与高频特判模式对应的硬件加速执行方式,在片上存储范围内完成张量维度重组;当前维度重组操作未匹配高频特判模式时,采用面向层次化存储结构的通用重组执行方式,通过并行计算与协同处理完成张量维度重组。能够提升reshape执行效率,降低片上内存资源的占用,避免带宽资源浪费。
本发明授权用于张量处理单元的张量维度重组方法及装置、芯片在权利要求书中公布了:1.一种用于张量处理单元的张量维度重组方法,其特征在于,张量处理单元具有层次化存储结构和向量并行计算单元,所述方法包括: 对于所述张量处理单元中待处理的输入张量,校验输入张量与目标输出张量的元素总数是否相等,以满足张量维度重组的基本约束; 基于输入张量和目标输出张量的形状参数、存储布局信息以及所述张量处理单元的硬件架构特性,对当前维度重组操作进行模式判定,以判断当前维度重组操作是否匹配预设的高频特判模式,包括:获取输入张量与目标输出张量的各维度尺寸及维度排列顺序,判断输入张量与目标输出张量之间是否满足预设的维度一一对应关系或维度互换关系;和或,判断输入张量和或目标输出张量中是否存在维度尺寸为预设阈值的单维度结构,并基于所述单维度结构在张量中的位置判断是否满足高频特判模式的形状约束条件;和或,获取输入张量与目标输出张量的存储布局信息,所述存储布局信息至少包括存储偏移、填充方式及连续性标识,并基于所述存储布局信息判断输入张量与目标输出张量是否在物理存储布局上保持连续一致;和或,判断输入张量与目标输出张量的数据类型是否一致,且是否满足所述张量处理单元的向量指令所要求的数据宽度和对齐条件;和或,判断是否属于张量处理单元支持的可直接重组的数据类型集合;和或,当至少一项判定结果满足高频特判模式对应的触发条件时,将当前维度重组操作判定为匹配所述高频特判模式; 当前维度重组操作匹配所述高频特判模式时,采用与所述高频特判模式对应的硬件加速执行方式,在片上存储范围内完成张量维度重组,得到目标输出张量; 当前维度重组操作未匹配所述高频特判模式时,采用面向层次化存储结构的通用重组执行方式,通过并行计算与协同处理完成张量维度重组,得到目标输出张量。
如需购买、转让、实施、许可或投资类似专利技术,可联系本专利的申请人或专利权人中昊芯英(杭州)科技有限公司,其通讯地址为:310052 浙江省杭州市滨江区长河街道聚才路239号6幢19层;或者联系龙图腾网官方客服,联系龙图腾网可拨打电话0551-65771310或微信搜索“龙图腾网”。
以上内容由龙图腾AI智能生成。
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。

皖公网安备 34010402703815号
请提出您的宝贵建议,有机会获取IP积分或其他奖励