검색으로 돌아가기
기록
一种面向异构计算平台的深度学习框架跨平台移植方法
발명유효
2조회수
7청구항 · 1 독립항
§ Ⅰ
개요
발명자
陈浩楠; 曾艳; 王珏; 周纯葆; 曹荣强; 王子鉴
IPC 분류
G06N 3/10 (2006.01)G06F 8/76 (2018.01)G06F 8/30 (2018.01)G06F 11/3668 (2025.01)
一种面向异构计算平台的深度学习框架跨平台移植方法,包括:构建分层式硬件抽象层和动态运行时选择器;构建高性能模板库;对性能敏感型算子手动编写模板特化版本;为每个手动优化的模板定义一个规范的配置接口,明确其约束条件;构建智能脚本生成器,并遍历目标平台整个框架的算子列表;对于每个需要移植的算子,在高性能模板库中查找匹配的模板;若找到,根据目标平台参数自动实例化该模板,生成对应的手动源文件;若未找到,调用通用模板引擎的通用代码生成模块,根据预设通用规则生成自动实现的源文件;智能脚本生成器自动生成适配目标平台的CMakeLists.txt文件,正确链接所有生成的手动源文件、自动实现的源文件和目标平台的底层驱动库;对目标平台进行测试。