[发明专利]一种在TVM中实现TensorCore卷积计算的方法及系统在审
| 申请号： | 202010663682.X | 申请日： | 2020-07-10 |
| 公开（公告）号： | CN111949932A | 公开（公告）日： | 2020-11-17 |
| 发明（设计）人： | 刘文川 | 申请（专利权）人： | 苏州浪潮智能科技有限公司 |
| 主分类号： | G06F17/15 | 分类号： | G06F17/15;G06F17/16 |
| 代理公司： | 济南诚智商标专利事务所有限公司 37105 | 代理人： | 王申雨 |
| 地址： | 215100 江苏省苏州市吴*** | 国省代码： | 江苏;32 |
| 权利要求书： | 查看更多 | 说明书： | 查看更多 |
| 摘要： | 本申请公开了一种在TVM中实现TensorCore卷积计算的方法及系统，该方法包括：获取TensorCore卷积输入数据和卷积核的数据布局；判断该数据布局是否满足：内部矩阵M、N、K维度上的大小是(16，16，16)，(8，32，16)或者(32，8，16)的倍数；如果是，判断输入数据的数据类型是否一致且均为int8或uint8；如果是，定义Tensorcore int8的卷积计算规则；执行调度优化设计；根据调度优化设计的结果，进行int8的TensorCore卷积计算。该系统包括：数据布局获取模块、第一判断模块、第二判断模块、int8卷积计算规则定义模块、float16卷积计算规则定义模块、设计模块和计算模块。通过本申请，能够卷积加速比达到1.5‑2左右，从而在AI图像识别网络模型的训练和推理过程中提高卷积计算性能。 | ||
| 搜索关键词： | 一种 tvm 实现 tensorcore 卷积 计算 方法 系统 | ||
【主权项】：
暂无信息
下载完整专利技术内容需要扣除积分，VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息，商用须获得专利权人授权。该专利全部权利属于苏州浪潮智能科技有限公司，未经苏州浪潮智能科技有限公司许可，擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作，请联系【客服】
本文链接：http://www.vipzhuanli.com/patent/202010663682.X/，转载请声明来源钻瓜专利网。
- 上一篇：一种直流电缆故障类型和位置判断方法及系统
- 下一篇：一种采矿用矿石破碎装置
