← 返回列表
用于计算的方法、计算装置、介质和程序产品
摘要文本
本发明涉及一种用于计算的方法、计算装置、介质和程序产品。该方法包括:基于静态形状,生成核函数;将核函数的各层循环代码拆分为多段分代码,以便对多段分代码所包括的主体分代码进行循环展开优化;以及经由寄存器,动态控制主体分代码的循环次数,以便基于静态形状所生成的核函数支持动态维度。本发明不仅能够有效支持动态形状,而且能够避免访存和算力的浪费,提高执行性能。
申请人信息
- 申请人:北京壁仞科技开发有限公司; 上海壁仞科技股份有限公司
- 申请人地址:100102 北京市朝阳区望京东园四区13号楼-4至33层101内10层201室
- 发明人: 北京壁仞科技开发有限公司; 上海壁仞科技股份有限公司
专利详细信息
| 项目 | 内容 |
|---|---|
| 专利名称 | 用于计算的方法、计算装置、介质和程序产品 |
| 专利类型 | 发明申请 |
| 申请号 | CN202410072686.9 |
| 申请日 | 2024/1/18 |
| 公告号 | CN117591776A |
| 公开日 | 2024/2/23 |
| IPC主分类号 | G06F17/10 |
| 权利人 | 北京壁仞科技开发有限公司; 上海壁仞科技股份有限公司 |
| 发明人 | 请求不公布姓名; 请求不公布姓名; 请求不公布姓名; 请求不公布姓名; 请求不公布姓名; 请求不公布姓名; 请求不公布姓名; 请求不公布姓名 |
| 地址 | 北京市朝阳区望京东园四区13号楼-4至33层101内10层201室; 上海市闵行区陈行公路2388号16幢13层1302室 |
专利主权项内容
1.一种用于计算的方法,其特征在于,所述方法包括:基于静态形状,生成核函数;将核函数的各层循环代码拆分为多段分代码,以便对多段分代码所包括的主体分代码进行循环展开优化;以及经由寄存器,动态控制主体分代码的循环次数,以便基于静态形状所生成的核函数支持动态维度。