← 返回列表

用于计算的方法、计算装置、介质和程序产品

申请号: CN202410072686.9
申请人: 北京壁仞科技开发有限公司; 上海壁仞科技股份有限公司
申请日期: 2024/1/18

摘要文本

本发明涉及一种用于计算的方法、计算装置、介质和程序产品。该方法包括:基于静态形状,生成核函数;将核函数的各层循环代码拆分为多段分代码,以便对多段分代码所包括的主体分代码进行循环展开优化;以及经由寄存器,动态控制主体分代码的循环次数,以便基于静态形状所生成的核函数支持动态维度。本发明不仅能够有效支持动态形状,而且能够避免访存和算力的浪费,提高执行性能。

专利详细信息

项目 内容
专利名称 用于计算的方法、计算装置、介质和程序产品
专利类型 发明申请
申请号 CN202410072686.9
申请日 2024/1/18
公告号 CN117591776A
公开日 2024/2/23
IPC主分类号 G06F17/10
权利人 北京壁仞科技开发有限公司; 上海壁仞科技股份有限公司
发明人 请求不公布姓名; 请求不公布姓名; 请求不公布姓名; 请求不公布姓名; 请求不公布姓名; 请求不公布姓名; 请求不公布姓名; 请求不公布姓名
地址 北京市朝阳区望京东园四区13号楼-4至33层101内10层201室; 上海市闵行区陈行公路2388号16幢13层1302室

专利主权项内容

1.一种用于计算的方法,其特征在于,所述方法包括:基于静态形状,生成核函数;将核函数的各层循环代码拆分为多段分代码,以便对多段分代码所包括的主体分代码进行循环展开优化;以及经由寄存器,动态控制主体分代码的循环次数,以便基于静态形状所生成的核函数支持动态维度。