sKokkos: Enabling Kokkos with Transparent Device Selection on Heterogeneous Systems using OpenACC

Conference · Sun Dec 31 23:00:00 EST 2023

OSTI ID:2438755

Lee, Seyong ^[1]; Vetter, Jeffrey ^[1]; Denny, Joel ^[1]; Teranishi, Keita ^[1]; Valero Lara, Pedro ^[1]

---

1. ORNL

This paper presents a new feature to enable Kokkos with transparent device selection. For application developers, it is not easy toidentify which device is the most appropriate to use in a heterogeneous system, since this depends on the characteristics of both the application and the hardware. In Kokkos, a backend is associated with one specific programming model/hardware. Programmers decide which backend to use at compilation time. This new feature implemented on the OpenACC backend eliminates the burden of deciding which device to use, providing a highly productive programming solution for Kokkos applications. This work includes implementation details and a performance study conducted with a set of mini-benchmarks (i.e., AXPY and dot product), kernels (Lattice-Bolzmann method), and two mini-apps (LULESH and miniFE) on two heterogeneous systems with different hardware capabilities. This new Kokkos feature provides high accelerations of up to 35× thanks to automatic and transparent device selection.

• Research Organization: Oak Ridge National Laboratory (ORNL), Oak Ridge, TN (United States)
• Sponsoring Organization: USDOE
• DOE Contract Number: AC05-00OR22725
• OSTI ID: 2438755
• Country of Publication: United States
• Language: English