[发明专利]一种基于目标众核的对抗网络移植与优化的方法及系统有效
申请号: | 202010306980.3 | 申请日: | 2020-04-17 |
公开(公告)号: | CN111666071B | 公开(公告)日: | 2021-12-17 |
发明(设计)人: | 魏志强;刘园园;杨永全 | 申请(专利权)人: | 中国海洋大学;青岛海洋科学与技术国家实验室发展中心 |
主分类号: | G06F8/30 | 分类号: | G06F8/30;G06F8/41;G06F11/36;G06N3/063;G06N3/08 |
代理公司: | 北京工信联合知识产权代理有限公司 11266 | 代理人: | 姜丽楼 |
地址: | 266100 山*** | 国省代码: | 山东;37 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 一种 基于 目标 对抗 网络 移植 优化 方法 系统 | ||
1.一种基于目标众核的对抗网络移植与优化的方法,所述方法包括:
编写对抗网络代码并移植到目标众核;
编写基于所述目标众核的对抗网络编译文件;
通过所述目标众核主核串行方式运行对抗网络的功能模型,对所述功能模型进行分析,对所述功能模型的热点函数进行定位,获取所述热点函数的统计结果;
根据所述热点函数的统计结果,对所述热点函数的特征和参数进行分析,编写所述热点函数的主核加速代码和所述热点函数的从核加速代码;
通过加速库对所述热点函数的主核加速代码和所述热点函数的从核加速代码的加速可行性进行测试;
当所述热点函数的主核加速代码和所述热点函数的从核加速代码的加速可行性通过测试时,通过所述加速库生成所述功能模型的优化策略,所述优化策略包括:DMA与从核cache结合的优化策略、DMA与RMA共同广播传输数据的优化策略。
2.根据权利要求1所述的方法,所述编写对抗网络代码并移植到目标众核;编写基于所述目标众核的对抗网络编译文件,还包括:
根据所述对抗网络的算法,编写适应于所述目标众核的代码并移植到目标众核;
根据所述目标众核的编译指令,编写基于所述目标众核的对抗网络编译文件,所述编译文件通过主核串行方式运行。
3.根据权利要求1所述的方法,所述功能模型包括:图像合成模型、文本预测模型以及图像修复模型。
4.根据权利要求1所述的方法,还包括:
通过加速库对所述热点函数的主从核加速代码的加速可行性进行测试;
计算所述热点函数的主从核加速代码的加速比,当计算出的加速比达到预设的阈值时,所述热点函数的主从核加速代码的加速可行性通过测试。
5.根据权利要求1所述的方法,还包括:
输入热点函数的代码;
挖掘所述热点函数功能与所述热点函数需求之间的关系;
对热点函数并行加速后的运行时间进行优化;
输出所述热点函数的主核代码和所述热点函数的从核代码;
判断所述热点函数的主核代码和所述热点函数的从核代码是否为最优化;
当所述热点函数的主核代码和所述热点函数的从核代码为最优化时,输入最优的热点函数的代码。
6.一种基于目标众核的对抗网络移植与优化的系统,所述系统包括:
第一编译单元,用于编写对抗网络代码并移植到目标众核;编写基于所述目标众核的对抗网络编译文件;
分析单元,用于通过所述目标众核主核串行方式运行对抗网络的功能模型,对所述功能模型进行分析,对所述功能模型的热点函数进行定位,获取所述热点函数的统计结果;
第二编译单元,用于根据所述热点函数的统计结果,对所述热点函数的特征和参数进行分析,编写所述热点函数的主核加速代码和所述热点函数的从核加速代码;
测试单元,用于通过加速库对所述热点函数的主核加速代码和所述热点函数的从核加速代码的加速可行性进行测试;
生成单元,用于当所述热点函数的主核加速代码和所述热点函数的从核加速代码的加速可行性通过测试时,通过所述加速库生成所述功能模型的优化策略,所述优化策略包括:DMA与从核cache结合的优化策略、DMA与RMA共同广播传输数据的优化策略。
7.根据权利要求6所述的系统,所述第一编译单元用于编写对抗网络代码并移植到目标众核;编写基于所述目标众核的对抗网络编译文件,包括:
根据所述对抗网络的算法,编写适应于所述目标众核的代码并移植到目标众核;
根据所述目标众核的编译指令,编写基于所述目标众核的对抗网络编译文件,所述编译文件通过主核串行方式运行。
8.根据权利要求6所述的系统,所述功能模型包括:图像合成模型、文本预测模型以及图像修复模型。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于中国海洋大学;青岛海洋科学与技术国家实验室发展中心,未经中国海洋大学;青岛海洋科学与技术国家实验室发展中心许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202010306980.3/1.html,转载请声明来源钻瓜专利网。