Small cleanup
This commit is contained in:
@ -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;
|
||||||
|
}
|
||||||
|
|||||||
@ -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:
|
||||||
|
|
||||||
};
|
};
|
||||||
|
|||||||
@ -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;
|
||||||
}
|
}
|
||||||
|
|||||||
@ -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;
|
|
||||||
}
|
|
||||||
|
|||||||
@ -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
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user