龙图腾网&IPTOP
- 微信扫码登录
- 账号密码登录
- 短信登录/注册
申请/专利权人：上海燧原科技股份有限公司
申请日：2024-03-21
公开（公告）日：2024-06-07
公开（公告）号：CN118151896A
主分类号：G06F8/20
分类号：G06F8/20;G06F18/214;G06N3/08
优先权：
专利状态码：在审-实质审查的生效
法律状态：2024.06.25#实质审查的生效;2024.06.07#公开
摘要：本发明公开了一种加速卡内存峰值预测方法、装置、设备及存储介质，包括：获取目标加速卡对应的模型数据集，将模型数据集加载至目标芯片的高级缓存区中；其中，模型数据集中包括多个训练样本，各训练样本分别包括不同AI模型对应的模型结构数据、训练参数、代码数据以及目标加速卡的内存峰值；将高级缓存区中的各训练样本，逐个搬运至目标芯片的低级缓存区，并将低级缓存区中的各训练样本逐个搬运至计算单元；通过计算单元，使用各训练样本对多层神经网络模型进行迭代训练，得到目标加速卡对应的内存峰值预测模型。本发明实施例的技术方案可以加快AI模型代码优化，提高AI模型开发效率。
主权项：1.一种加速卡内存峰值预测方法，其特征在于，所述方法包括：获取目标加速卡对应的模型数据集，将所述模型数据集加载至目标芯片的高级缓存区中；其中，所述模型数据集中包括多个训练样本，各训练样本分别包括不同AI模型对应的模型结构数据、训练参数、代码数据以及目标加速卡的内存峰值；将所述高级缓存区中的各训练样本，逐个搬运至目标芯片的低级缓存区，并将所述低级缓存区中的各训练样本，逐个搬运至计算单元；通过所述计算单元，使用各所述训练样本对多层神经网络模型进行迭代训练，得到目标加速卡对应的内存峰值预测模型。
全文数据：
权利要求：
百度查询： 上海燧原科技股份有限公司 一种加速卡内存峰值预测方法、装置、设备及存储介质
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息，力求客观、公正，但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解，仅供参考使用，不能作为本公司承担任何法律责任的依据或者凭证。
主营中国专利交易买卖与专利转让许可,高校科技成果推广与科技成果转化,知识产权运营管理与平台开发,科技人才专家库与科技猎头等科技服务
开放平台
担保交易
惠及多方
会员特惠
专属平台
适合多方
数据共享
功能齐全
皖公网安备 34010402703815号