| 引用本文: | 沈莉,周文浩,王飞,李斌,谭坚,商红慧,安虹,漆锋滨.swJulia: 面向新一代神威超级计算机的Julia语言编译系统.软件学报,2025,36(12):5402-5422 |
| |
|
| 本文已被：浏览 1092次 下载 2618次 |
码上扫一扫！ |
|
|
| swJulia: 面向新一代神威超级计算机的Julia语言编译系统 |
|
沈莉1, 周文浩1, 王飞2, 李斌3, 谭坚1, 商红慧1, 安虹1, 漆锋滨3
|
|
1.中国科学技术大学, 安徽 合肥 230026;2.清华大学, 北京 100084;3.江南计算技术研究所, 江苏 无锡 214083
|
|
| 摘要: |
| 随着异构融合体系结构在高性能计算领域的普及, 挖掘其潜能并探索新的应用构建策略变得至关重要. 传统的静态编译方法已无法满足复杂计算需求, 动态编程语言因其灵活性和高效性而备受瞩目. Julia是一种现代的高性能动态编程语言, 其基于即时编译机制, 在科学计算等领域表现出色. 结合申威异构众核架构特点, 构建ORCJIT编译引擎并提出了动态模式下的片上存储管理方法, 并以此为基础实现针对新一代神威超级计算机的Julia动态语言编译器swJulia. 其不仅继承了Julia编译器的灵活性, 同时还有效支持了SACA众核编程模型及运行时封装. 利用swJulia编译系统, 成功在新一代神威超级计算机上部署了NNQS-Transformer量子化学模拟器, 并在多个维度验证了swJulia的好用性和高效性. 实验结果显示, swJulia在单线程基准测试和众核加速上性能卓越, 并能够有效支撑NNQS-Transformer量子化学模拟器的超大规模可扩展并行模拟. |
| 关键词: Julia编译器 神威超级计算机 动态链接 即时编译 量子化学模拟 |
| DOI：10.13328/j.cnki.jos.007407 |
| 分类号:TP314 |
| 基金项目:国家重点研发计划(2023YFB3001500) |
|
| swJulia: Julia Compilation System for New Generation Sunway Supercomputer |
|
SHEN Li1, ZHOU Wen-Hao1, WANG Fei2, LI Bin3, TAN Jian1, SHANG Hong-Hui1, AN Hong1, QI Feng-Bin3
|
|
1.University of Science and Technology of China, Hefei 230026, China;2.Tsinghua University, Beijing 100084, China;3.Jiangnan Institute of Computing Technology, Wuxi 214083, China
|
| Abstract: |
| With the increasing adoption of heterogeneous integrated architectures in high-performance computing, it has become essential to harness their potential and explore new strategies for application development. Traditional static compilation methodologies are no longer sufficient to meet the complex computational demands. Therefore, dynamic programming languages, known for their flexibility and efficiency, are gaining prominence. Julia, a modern high-performance language characterized by its JIT compilation mechanism, has demonstrated significant performance in fields such as scientific computing. Targeting the unique features of the Sunway heterogeneous many-core architecture, the ORCJIT engine is introduced, along with an on-chip storage management approach specifically designed for dynamic modes. Based on these advancements, swJulia is developed as a Julia dynamic language compiler tailored for the new generation of the Sunway supercomputer. This compiler not only inherits the flexibility of the Julia compiler but also provides robust support for the SACA many-core programming model and runtime encapsulation. By utilizing the swJulia compilation system, the deployment of the NNQS-Transformer quantum chemistry simulator on the new generation of the Sunway supercomputer is successfully achieved. Comprehensive validation across multiple dimensions demonstrates the efficacy and efficiency of swJulia. Experimental results show exceptional performance in single-threaded benchmark tests and many-core acceleration, significantly improving ultra-large-scale parallel simulations for the NNQS-Transformer quantum chemistry simulator. |
| Key words: Julia compiler Sunway supercomputer dynamic linking just-in-time compilation quantum chemistry simulation |
