Small cleanup

This commit is contained in:
Stan Moore
2023-04-28 14:57:19 -06:00
parent 235372d6e8
commit 29a2b58ab6
5 changed files with 25 additions and 36 deletions

View File

@ -394,7 +394,7 @@ void ModifyKokkos::final_integrate()
/* ---------------------------------------------------------------------- /* ----------------------------------------------------------------------
2nd half of integrate call, only for relevant fixes fused initial and final integrate call, only for relevant fixes
------------------------------------------------------------------------- */ ------------------------------------------------------------------------- */
void ModifyKokkos::fused_integrate() void ModifyKokkos::fused_integrate()
@ -900,3 +900,22 @@ int ModifyKokkos::min_reset_ref()
} }
return itmpall; return itmpall;
} }
/* ----------------------------------------------------------------------
check if initial and final integrate can be fused
------------------------------------------------------------------------- */
int ModifyKokkos::check_fuse_integrate()
{
int fuse_integrate_flag = 1;
for (int i = 0; i < n_initial_integrate; i++)
if (!fix[list_initial_integrate[i]]->fuse_integrate_flag)
fuse_integrate_flag = 0;
for (int i = 0; i < n_final_integrate; i++)
if (!fix[list_final_integrate[i]]->fuse_integrate_flag)
fuse_integrate_flag = 0;
return fuse_integrate_flag;
}

View File

@ -70,6 +70,8 @@ class ModifyKokkos : public Modify {
int min_dof() override; int min_dof() override;
int min_reset_ref() override; int min_reset_ref() override;
int check_fuse_integrate();
protected: protected:
}; };

View File

@ -27,7 +27,7 @@
#include "kspace.h" #include "kspace.h"
#include "output.h" #include "output.h"
#include "update.h" #include "update.h"
#include "modify.h" #include "modify_kokkos.h"
#include "timer.h" #include "timer.h"
#include "memory_kokkos.h" #include "memory_kokkos.h"
#include "kokkos.h" #include "kokkos.h"
@ -629,5 +629,5 @@ void VerletKokkos::fuse_check(int i, int n)
if (i == 0 || i == n-1) fuse_integrate = 0; if (i == 0 || i == n-1) fuse_integrate = 0;
if (update->ntimestep == output->next) fuse_integrate = 0; if (update->ntimestep == output->next) fuse_integrate = 0;
if (timer->has_timeout()) fuse_integrate = 0; if (timer->has_timeout()) fuse_integrate = 0;
if (!modify->check_fuse_integrate()) fuse_integrate = 0; if (!((ModifyKokkos*)modify)->check_fuse_integrate()) fuse_integrate = 0;
} }

View File

@ -475,17 +475,6 @@ void Modify::final_integrate()
for (int i = 0; i < n_final_integrate; i++) fix[list_final_integrate[i]]->final_integrate(); for (int i = 0; i < n_final_integrate; i++) fix[list_final_integrate[i]]->final_integrate();
} }
/* ----------------------------------------------------------------------
2nd half of integrate call, only for relevant fixes
------------------------------------------------------------------------- */
void Modify::fused_integrate()
{
for (int i = 0; i < n_final_integrate; i++)
fix[list_final_integrate[i]]->fused_integrate();
}
/* ---------------------------------------------------------------------- /* ----------------------------------------------------------------------
end-of-timestep call, only for relevant fixes end-of-timestep call, only for relevant fixes
only call fix->end_of_step() on timesteps that are multiples of nevery only call fix->end_of_step() on timesteps that are multiples of nevery
@ -1810,22 +1799,3 @@ double Modify::memory_usage()
for (int i = 0; i < ncompute; i++) bytes += compute[i]->memory_usage(); for (int i = 0; i < ncompute; i++) bytes += compute[i]->memory_usage();
return bytes; return bytes;
} }
/* ----------------------------------------------------------------------
check if initial and final integrate can be fused
------------------------------------------------------------------------- */
int Modify::check_fuse_integrate()
{
int fuse_integrate_flag = 1;
for (int i = 0; i < n_initial_integrate; i++)
if (!fix[list_initial_integrate[i]]->fuse_integrate_flag)
fuse_integrate_flag = 0;
for (int i = 0; i < n_final_integrate; i++)
if (!fix[list_final_integrate[i]]->fuse_integrate_flag)
fuse_integrate_flag = 0;
return fuse_integrate_flag;
}

View File

@ -61,7 +61,7 @@ class Modify : protected Pointers {
virtual void setup_pre_force(int); virtual void setup_pre_force(int);
virtual void setup_pre_reverse(int, int); virtual void setup_pre_reverse(int, int);
virtual void initial_integrate(int); virtual void initial_integrate(int);
virtual void fused_integrate(); virtual void fused_integrate() {}
virtual void post_integrate(); virtual void post_integrate();
virtual void pre_exchange(); virtual void pre_exchange();
virtual void pre_neighbor(); virtual void pre_neighbor();
@ -151,8 +151,6 @@ class Modify : protected Pointers {
double memory_usage(); double memory_usage();
int check_fuse_integrate();
protected: protected:
// internal fix counts // internal fix counts