我校学子首次参加第五届“神威杯“国产CPU并行应用挑战赛斩获银奖
发布人:华夏 上传时间:2021-10-25 阅读次数:

9月24-25日,第五届“神威杯”国产CPU并行应用挑战赛总决赛在无锡超算中心举行,并于10.21在珠海CCF HPC China 2021会议举行了颁奖仪式。我校计算机科学与技术学院三名本科生(19级本科生孙泽田、陈爵豪,20级本科生满洋)组成的参赛队伍“HITsz666”队伍总分排名第二,夺得银奖。实验与创新实践教育中心郑海刚任指导老师。

 

队伍比赛现场合影


在赛题设置上,大赛紧扣Transformer这一深度学习前沿网络和相关应用热点,要求参赛选手在新一代申威众核架构下,优化基于PyTorch实现的ViT模型训练任务,并针对其中的Transformer结构进行性能优化。 

在赛制环节上,大赛组委会不仅设计了面向Transformer核心结构的算子级优化Case,还有面向ViT模型全流程训练的任务级优化Case,为选手提供了一个多角度、多层次的优化技能竞技场。同时,本次大赛从算子输出结果的浮点数计算准确性和模型训练结果的收敛性对选手的提交结果进行验证,确保这场终极“大考”在保持对抗强度的基础上,让所有“芯”生代选手拥有一个公平公正的竞技环境。

在历时近一个月的决赛中,我校“HITsz666“队伍通过分析ViT模型的结构和计算热点,结合SW26010PRO处理器的架构特点和性能瓶颈,采用了从核移植、负载均衡、缓冲优化、汇编改写、向量化计算和融合乘加等方法,累计实现了12种适用于不同环境的矩阵乘法,在“神威·太湖之光”国产超算平台上实现了对Vision Transformer计算速度的大幅优化。

 

队伍答辩现场


国产CPU并行应用挑战赛(China Parallel application Challenge on domestic CPU, 简称CPC)创始于2017年。五年来,CPC已累计覆盖全国23个省市(直辖市),先后将350所高校、科研单位的1000余支队伍与“超算”紧密联系在一起。比赛旨在更好地利用自主可控国产CPU在高性能计算领域的性能优势,进一步完善其软件生态环境,推动国产超算平台的产业化进程,通过比赛激励学术界和产业界积极参与国产CPU应用的开发和创新,提升我国高性能计算的整体水平。

 

获奖队伍名单

 

颁奖合影

最新发布