logo好方法网

一种VASP的资源调用方法、系统、设备以及介质


技术摘要:
本发明公开了一种VASP的资源调用方法,包括以下步骤:获取VASP所在平台的GPU的架构信息以及待调用GPU的数量;根据架构信息以及数量对平台进行环境初始化;获取GPU的互联方式以及GPU与CPU的绑定关系;按照互联方式的优先级顺序从高到低依次调用对应的GPU,直到调用的GPU  全部
背景技术:
一些科学计算软件,例如VASP、Lammps等已经通过编程人员的努力,利用CUDA编程 语言成功移植到了Nvidia的GPU显卡上。测试结果表明,当VASP软件在以CPU GPU混合异构 超级计算机上运行的时候,得到了极大的性能提升,但是无法实现对CPU和GPU的最优调度。
技术实现要素:
有鉴于此,为了克服上述问题的至少一个方面,本发明实施例提出一种VASP的资 源调用方法,包括以下步骤: 获取VASP所在平台的GPU的架构信息以及待调用GPU的数量; 根据所述架构信息以及所述数量对所述平台进行环境初始化; 获取所述GPU的互联方式以及所述GPU与CPU的绑定关系; 按照所述互联方式的优先级顺序从高到低依次调用对应的GPU,直到调用的GPU的 数量达到所述待调用GPU的数量; 根据所述绑定关系确定调用的GPU绑定的CPU的数量以及根据所述待调用GPU的数 量确定待调用CPU的数量; 响应于所述待调用CPU的数量小于所述绑定的CPU的数量,从所述绑定的CPU中调 用对应数量的所述待调用CPU。 在一些实施例中,还包括: 响应于所述待调用CPU的数量不小于所述绑定的CPU的数量,调用所有所述绑定的 CPU,并调用其他未与所述调用的GPU有绑定关系的CPU,直到调用的CPU的数量达到所述待 调用CPU的数量。 在一些实施例中,还包括: 比较所述VASP的输入文件中的IBZKPT参数与所述调用的GPU的数量的大小; 响应于所述IBZKPT参数不小于所述调用的GPU的数量,将所述KPAR参数调整为与 所述调用的GPU的数量相等; 响应于所述IBZKPT参数小于所述调用的GPU的数量,判断所述IBZKPT参数奇偶性; 响应于所述IBZKPT参数为奇数,将所述KPAR参数调整为所述调用的GPU的数量的 二分之一; 响应于所述IBZKPT参数为偶数,将所述KPAR参数调整为与所述IBZKPT参数相等。 在一些实施例中,还包括: 将NCORE参数调整为1。 在一些实施例中,按照所述互联方式的优先级顺序从高到低依次调用对应的GPU, 4 CN 111580976 A 说 明 书 2/8 页 直到调用的GPU的数量达到所述待调用GPU的数量,进一步包括: 开启所述调用的GPU持续性模式以及多进程服务; 根据每一个所述调用的GPU的底层信息分别将所述每一个调用的GPU的时钟频率 调节至最大值。 在一些实施例中,获取VASP所在平台的GPU的架构信息和待调用GPU的数量,进一 步包括: 获取所有的GPU的数量、GPU的计算能力、CUDA版本、CUDA数学库绝对路径。 在一些实施例中,还包括: 判断是否接收到用户输入的GPU的数量; 响应于接收到所述用户输入的GPU的数量,将所述用户输入的GPU的数量作为所述 待调用GPU的数量; 响应于没有接收到所述用户输入的GPU的数量,将所述所有的GPU的数量作为所述 待调用GPU的数量。 基于同一发明构思,根据本发明的另一个方面,本发明的实施例还提供了一种 VASP的资源调用系统,包括: 第一获取模块,所述第一获取模块配置为获取VASP所在平台的GPU的架构信息以 及待调用GPU的数量; 初始化模块,所述初始化模块配置为根据所述架构信息以及所述数量对所述平台 进行环境初始化; 第二获取模块,所述第二获取模块配置为获取所述GPU的互联方式以及所述GPU与 CPU的绑定关系; GPU调用模块,所述GPU调用模块配置为按照所述互联方式的优先级顺序从高到低 依次调用对应的GPU,直到调用的GPU的数量达到所述待调用GPU的数量; 确定模块,所述确定模块配置为根据所述绑定关系确定调用的GPU绑定的CPU的数 量以及根据所述待调用GPU的数量确定待调用CPU的数量; 第一响应模块,所述第一响应模块配置为响应于所述待调用CPU的数量小于所述 绑定的CPU的数量,从所述绑定的CPU中调用对应数量的所述待调用CPU。 基于同一发明构思,根据本发明的另一个方面,本发明的实施例还提供了一种计 算机设备,包括: 至少一个处理器;以及 存储器,所述存储器存储有可在所述处理器上运行的计算机程序,其特征在于,所 述处理器执行所述程序时执行如上所述的任一种VASP的资源调用方法的步骤。 基于同一发明构思,根据本发明的另一个方面,本发明的实施例还提供了一种计 算机可读存储介质,所述计算机可读存储介质存储有计算机程序,所述计算机程序被处理 器执行时执行如上所述的任一种VASP的资源调用方法的步骤。 本发明具有以下有益技术效果之一:本发明提出的方案能够极大节约了软件重新 部署的时间以及使用者对于硬件平台的学习时间成本,并且实现了在运行VASP软件时,CPU 和GPU资源的最优调度。 5 CN 111580976 A 说 明 书 3/8 页 附图说明 为了更清楚地说明本发明实施例或现有技术中的技术方案,下面将对实施例或现 有技术描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图仅仅是本 发明的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以 根据这些附图获得其他的实施例。 图1为本发明的实施例提供的VASP的资源调用方法的流程示意图; 图2为本发明的实施例提供的VASP的资源调用系统的结构示意图; 图3为本发明的实施例提供的计算机设备的结构示意图; 图4为本发明的实施例提供的计算机可读存储介质的结构示意图。
分享到:
收藏