support running with Kokkos + OpenMP

This commit is contained in:
Axel Kohlmeyer
2023-08-02 21:06:30 -04:00
parent 07b36025e2
commit f6732f028c
4 changed files with 29 additions and 2 deletions

View File

@ -159,7 +159,16 @@ AcceleratorTab::AcceleratorTab(QSettings *_settings, LammpsWrapper *_lammps, QWi
openmp->setObjectName("openmp");
intel->setEnabled(lammps->config_has_package("INTEL"));
intel->setObjectName("intel");
kokkos->setEnabled(lammps->config_has_package("KOKKOS"));
// Kokkos support only works with OpenMP for now.
if (lammps->config_has_package("KOKKOS")) {
if (lammps->config_accelerator("KOKKOS", "api", "openmp") &&
!(lammps->config_accelerator("KOKKOS", "api", "cuda") ||
lammps->config_accelerator("KOKKOS", "api", "hip") ||
lammps->config_accelerator("KOKKOS", "api", "sycl")))
kokkos->setEnabled(true);
else
kokkos->setEnabled(false);
}
kokkos->setObjectName("kokkos");
gpu->setEnabled(lammps->config_has_package("GPU") && lammps->has_gpu_device());
gpu->setObjectName("gpu");