diff --git a/src/pair.cpp b/src/pair.cpp index b8331086f7..dde19c1d05 100644 --- a/src/pair.cpp +++ b/src/pair.cpp @@ -30,6 +30,7 @@ #include "comm.h" #include "force.h" #include "update.h" +#include "accelerator.h" #include "memory.h" #include "error.h" @@ -40,6 +41,7 @@ using namespace LAMMPS_NS; enum{GEOMETRIC,ARITHMETIC,SIXTHPOWER}; enum{R,RSQ,BMP}; +enum{NOACCEL,OPT,GPU,USERCUDA}; // same as lammps.cpp /* ---------------------------------------------------------------------- */ @@ -325,6 +327,9 @@ void Pair::ev_setup(int eflag, int vflag) if (vflag_atom == 0) vflag_either = 0; if (vflag_either == 0 && eflag_either == 0) evflag = 0; } else vflag_fdotr = 0; + + if (lmp->accelerator == USERCUDA) + lmp->cuda->evsetup_eatom_vatom(eflag_atom,vflag_atom); } /* ----------------------------------------------------------------------