Improve DLB+PME tuning with GPUs

[alexxy/gromacs.git] / src / gromacs / ewald / pme-load-balancing.h

diff --git a/src/gromacs/ewald/pme-load-balancing.h b/src/gromacs/ewald/pme-load-balancing.h
index 2a46b2cb1ee2d2560f0c68d7dabeec11c9b7cb47..4b307cd76d1e8c49f581b2ae7db3c7ce73fd8367 100644 (file)
--- a/src/gromacs/ewald/pme-load-balancing.h
+++ b/src/gromacs/ewald/pme-load-balancing.h
@@ -65,12 +65,13 @@ struct pme_load_balancing_t;
  * usage.
  */
 void pme_loadbal_init(pme_load_balancing_t     **pme_lb_p,
+                      t_commrec                 *cr,
+                      FILE                      *fp_log,
                       const t_inputrec          *ir,
                       matrix                     box,
                       const interaction_const_t *ic,
                       struct gmx_pme_t          *pmedata,
                       gmx_bool                   bUseGPU,
-                      gmx_bool                   bSepPMERanks,
                       gmx_bool                  *bPrinting);
 
 /*! \brief Process cycles and PME load balance when necessary
@@ -92,12 +93,10 @@ void pme_loadbal_do(pme_load_balancing_t *pme_lb,
                     gmx_int64_t           step_rel,
                     gmx_bool             *bPrinting);
 
-/*! \brief Restart the PME load balancing discarding all timings gathered up till now */
-void restart_pme_loadbal(pme_load_balancing_t *pme_lb, int n);
-
 /*! \brief Finish the PME load balancing and print the settings when fplog!=NULL */
 void pme_loadbal_done(pme_load_balancing_t *pme_lb,
-                      t_commrec *cr, FILE *fplog,
-                      gmx_bool bNonBondedOnGPU);
+                      t_commrec            *cr,
+                      FILE                 *fplog,
+                      gmx_bool              bNonBondedOnGPU);
 
 #endif