diff --git a/mala/descriptors/descriptor.py b/mala/descriptors/descriptor.py index 0c055a4e0..3600d7e43 100644 --- a/mala/descriptors/descriptor.py +++ b/mala/descriptors/descriptor.py @@ -757,18 +757,18 @@ def _setup_lammps( lammps_dict["ngridy"] = ny lammps_dict["ngridz"] = nz lammps_dict["switch"] = self.parameters.bispectrum_switchflag - if self.parameters._configuration["gpu"]: - # Tell Kokkos to use one GPU. - lmp_cmdargs.append("-k") - lmp_cmdargs.append("on") - lmp_cmdargs.append("g") - lmp_cmdargs.append("1") - - # Tell LAMMPS to use Kokkos versions of those commands for - # which a Kokkos version exists. - lmp_cmdargs.append("-sf") - lmp_cmdargs.append("kk") - pass + if self.parameters._configuration["gpu"]: + # Tell Kokkos to use one GPU. + lmp_cmdargs.append("-k") + lmp_cmdargs.append("on") + lmp_cmdargs.append("g") + lmp_cmdargs.append("1") + + # Tell LAMMPS to use Kokkos versions of those commands for + # which a Kokkos version exists. + lmp_cmdargs.append("-sf") + lmp_cmdargs.append("kk") + pass lmp_cmdargs = set_cmdlinevars(lmp_cmdargs, lammps_dict)