千卡 GPU 集群上构建脉冲神经网络的新方法

脉冲神经网络（SNN）模拟涉及大规模延迟微分方程系统，在 HPC 集群上高效运行面临通信与内存管理的挑战。本文提出一种基于 MPI 的多 GPU 集群网络构建方法，每个进程独立构建本地连接并准备高效尖峰交换的数据结构，支持点对点与集合通信两种模式，在数千 GPU 上展示了两种皮层模型的扩展性能。