Porting codes to GPU often requires major efforts. While several tools exist for automatically offload numerical libraries such as BLAS and LAPACK, they often prove impractical due to the high cost of mandatory data transfer. The new unified memory architecture in NVIDIA Grace-Hopper allows high bandwidth cache-coherent memory access of all memory from both CPU and GPU, potentially eliminating bottleneck faced in conventional architecture. This breakthrough opens up new avenues for application development and porting strategies. In this study, we introduce a new tool for automatic BLAS offload, the tool leverages the high speed cache coherent NVLink C2C interconnect in Grace-Hopper, and enables performant GPU offload for BLAS heavy applications with no code changes or recompilation. The tool was tested on two quantum chemistry or physics codes, great performance benefits were observed.
翻译:将代码移植到GPU通常需要大量工作。尽管存在多种自动卸载数值库(如BLAS和LAPACK)的工具,但由于强制数据传输的高昂成本,这些工具往往不实用。NVIDIA Grace-Hopper的新型统一内存架构允许CPU和GPU对所有内存进行高带宽缓存一致性访问,从而可能消除传统架构中面临的瓶颈。这一突破为应用程序开发和移植策略开辟了新途径。在本研究中,我们引入了一种新的自动BLAS卸载工具,该工具利用Grace-Hopper的高速缓存一致性NVLink C2C互连,无需修改代码或重新编译即可实现针对BLAS密集型应用的高性能GPU卸载。该工具已在两个量子化学或物理学代码上进行了测试,并观察到了显著的性能提升。