← 返回列表
基于AI芯片的DMA带宽确定方法、装置、设备及介质
摘要文本
本发明公开了一种基于AI芯片的DMA带宽确定方法、装置、设备及介质。通过获取目标AI芯片对应的当前DMA数据；将当前DMA数据通过预先构建的数据预处理方法进行数据处理，得到标准的当前DMA数据；使用标准的当前DMA数据输入至预先训练完成的芯片带宽确定数学模型中，确定出目标AI芯片在不同数据搬运情况下所对应的DMA带宽；将DMA带宽进行反馈，以确定目标AI芯片对应的数据搬运中的所述DMA带宽最佳性能。解决了对不同AI芯片的DMA带宽推断和评估不准确的问题，实现了对AI芯片的极限估值，提高了对不同类型的AI芯片的DMA带宽推断的准确性。 微信公众号马克 数据网
申请人信息
- 申请人:上海燧原智能科技有限公司
- 申请人地址:201306 上海市浦东新区中国(上海)自由贸易试验区临港新片区业盛路188号A-522室
- 发明人: 上海燧原智能科技有限公司
专利详细信息
| 项目 | 内容 |
|---|---|
| 专利名称 | 基于AI芯片的DMA带宽确定方法、装置、设备及介质 |
| 专利类型 | 发明申请 |
| 申请号 | CN202410251615.5 |
| 申请日 | 2024/3/6 |
| 公告号 | CN117827710A |
| 公开日 | 2024/4/5 |
| IPC主分类号 | G06F13/28 |
| 权利人 | 上海燧原智能科技有限公司 |
| 发明人 | 朱佳乐; 王军伟; 孙诚程; 贾明桥 |
| 地址 | 上海市浦东新区中国(上海)自由贸易试验区临港新片区业盛路188号A-522室 |
专利主权项内容
1.一种基于AI芯片的DMA带宽确定方法，其特征在于，包括：获取目标AI芯片对应的当前DMA数据；其中，所述当前DMA数据是所述目标AI芯片中的最小颗粒度的直接使用DMA进行搬运的数据；将所述当前DMA数据通过预先构建的数据预处理方法进行数据处理，得到标准的当前DMA数据；使用所述标准的当前DMA数据输入至预先训练完成的芯片带宽确定数学模型中，确定出所述目标AI芯片在不同数据搬运情况下所对应的DMA带宽；将所述DMA带宽进行反馈，以确定所述目标AI芯片对应的数据搬运中的所述DMA带宽最佳性能。