·
Fast CUDA Graph Construction for ATLAS GNN Tracking
Graph construction was the slowest step of ATLAS's GNN track reconstruction. A CUDA port, organised as doublets first and then triplets, brought it from about ten minutes per event on a CPU to 69 ms on one GPU.
Key result
10⁴×
speedup over CPU
From about ten minutes per event on a CPU to 69 ms on one A100 GPU, the port that made GNN tracking fit the HL-LHC computing budget. Now part of ATLAS's GNN tracking pipeline for HL-LHC studies, and tuned further by the collaboration since.
Resources