Porting CMS Heterogeneous Pixel Reconstruction to Kokkos

Conference · Tue Apr 13 04:00:00 EDT 2021

OSTI ID:1778682

Childers, Taylor ^[1]; Kortelainen, Matti J. ^[2]; Kwok, Martin ^[2]; Strelchenko, Alexei ^[2]; Wang, Yunsong ^[3]

---

1. Argonne (main)
2. Fermilab
3. LBL, Berkeley

Programming for a diverse set of compute accelerators in addition to the CPU is a challenge. Maintaining separate source code for each architecture would require lots of effort, and development of new algorithms would be daunting if it had to be repeated many times. Fortunately there are several portability technologies on the market such as Alpaka, Kokkos, and SYCL. These technologies aim to improve the developer productivity by making it possible to use the same source code for many different architectures. In this paper we use heterogeneous pixel reconstruction code from the CMS experiment at the CERNL LHC as a realistic use case of a GPU-targeting HEP reconstruction software, and report experience from prototyping a portable version of it using Kokkos. The development was done in a standalone program that attempts to model many of the complexities of a HEP data processing framework such as CMSSW. We also compare the achieved event processing throughput to the original CUDA code and a CPU version of it.

• Research Organization: Lawrence Berkeley National Laboratory (LBNL), Berkeley, CA (United States); Fermi National Accelerator Laboratory (FNAL), Batavia, IL (United States)
• Sponsoring Organization: USDOE Office of Science (SC), High Energy Physics (HEP) (SC-25)
• DOE Contract Number: AC02-07CH11359
• OSTI ID: 1778682
• Report Number(s): FERMILAB-CONF-21-063-SCD; arXiv:2104.06573; oai:inspirehep.net:1858274
• Country of Publication: United States
• Language: English