GPU加速在第一性原理输运研究中的应用  

GPU Acceleration for First Principles Transport Calculations

作　　者：郑小宏[1] 兰杰[1] 郝华[1] 曾雉[1] 

机构地区：[1]中国科学院合肥物质科学研究院固体物理研究所 材料物理重点实验室,安徽合肥230031

出　　处：《科研信息化技术与应用》2013年第5期90-96,共7页E-science Technology & Application

基　　金：国家自然科学基金(批准号:11174289);中国科学院信息化专项超级计算重点应用示范项目(项目编号:XXH12503-02-03-02)的资助

摘　　要：分子器件中的第一性原理输运计算比普通的密度泛函计算要慢很多,其最根本原因在于密度矩阵计算方法的不同。本文将GPU加速应用于第一性原理的输运计算,重点实现相关矩阵运算的加速。测试结果表明,在单次迭代中,对于较大的体系,相对于调用MKL库,密度矩阵的运算速度在单个Tesla M2090可以提高一个数量级以上,在Tesla K20m上则可以提高20倍以上,从而取得了很好的加速效果,而且体系越大,加速效果越好。First principles transport calculations in large molecular devices are generally very slow, especially when compared with the density functional calculations for electronic structures. For transport calculations, the most time consuming part is the calculation of the density matrix. In this work, GPU acceleration is applied to the first principles transport calculations and we focus mainly on the speedup for the matrix operations. Tests show that, in a single iteration, there will be a speedup of more than one order of magnitude for large systems on a single Tesla M2090 and more than 20 times on a single Tesla K20m, compared with the MKL library. Thus a very good acceleration is achieved, and the bigger the system is, the better speedup we will achieve.

关 键 词：第一性原理输运计算 GPU加速 密度矩阵 矩阵运算 

分 类 号：O241.6[理学—计算数学]