国产高性能算力芯片研发

我方研究任务及考核指标

• 研究任务：研究在海光DPU大规模集群上的模型划分、内存优化、参数通信优化等核心技术。
• 考核指标：申请1个软件著作权，申请2项发明专利，发表2篇学术文章，培养2名硕士（或博士）研究生。

项目阶段及任务划分

第一阶段：方案设计（2024年9月~2024年12月）

• 完成硬件平台和框架的调研与分析。
• 确定移植与优化的技术路线和工作目标，针对气象大模型相关技术开展研究。

第二阶段：框架移植（2025年1月~2025年10月）

• 平台建设（前期通过平台租赁，后续采购海光硬件）。
• 实现智能模型训练框架的移植，完成基本的计算核心适配。
• 对计算进行初步优化，确保框架可以在国产芯片上运行。

第三阶段：性能优化（2025年11月~2026年12月）

• 开展模型训练在DPU平台上性能优化，完成数据并行、流水线并行、张量并行及通信组件优化工作。
• 完成全面的稳定性和兼容性测试，确保训练过程无误。
• 发表相关学术论文2篇，申请专利2项。

第四阶段：应用示范（2027年1月~2027年11月）

• 开发与国产芯片配套的工具链，提供调试和性能分析工具。
• 申请软著1项，培养硕士研究生2名。
• 在气象领域开展应用示范。
• 编写项目相关文档，包括使用手册和技术报告。