一种AI模型编译方法、设备及存储介质
基本信息
| 申请号 | CN201911040449.X | 申请日 | - |
| 公开（公告）号 | CN110750298B | 公开（公告）日 | 2021-03-09 |
| 申请公布号 | CN110750298B | 申请公布日 | 2021-03-09 |
| 分类号 | G06F8/76(2018.01)I;G06F8/41(2018.01)I | 分类 | 计算；推算；计数; |
| 发明人 | 夏正勋;罗圣美 | 申请（专利权）人 | 南京星环智能科技有限公司 |
| 代理机构 | 北京品源专利代理有限公司 | 代理人 | 孟金喆 |
| 地址 | 210019江苏省南京市建邺区贤坤路1号科创中心2楼220-351号 | ||
| 法律状态 | - | ||
摘要
| 摘要 | 本发明实施例公开了一种AI模型编译方法、设备及存储介质。其中，方法包括：获取通过标准AI框架训练得到的AI模型，并解析AI模型；对解析得到的AI模型中的算子进行分类，得到待适配的AI芯片支持的第一类算子集以及AI芯片不支持的第二类算子集；根据第一类算子集以及第二类算子集，生成至少一项中间表达；根据至少一项中间表达生成用于在所述AI芯片上运行的执行模型。本发明实施例的方案实现了将AI模型适配于不同的AI芯片，简化了AI模型的应用。 |
