This is partner PR to https://github.com/lattice/quda/pull/1014, which enables full residency of the momentum field during RHMC evolution avoiding copies between host and device.
Momentum action evaluation is now offloaded to QUDA if HAVE_QUDA is set.
When both the gauge and fermion forces are offloaded to QUDA, momentum residency is enabled. When the momentum heatbath is called in ranmom we download the momentum field, enabling residency using the qudaMomLoad QUDA interface function.
If the momentum is desired back on the host, then the function qudaMomSave can be used, which will invalidate the resident field, restoring host-device copying until qudaMomLoad is called again.
This is partner PR to https://github.com/lattice/quda/pull/1014, which enables full residency of the momentum field during RHMC evolution avoiding copies between host and device.
HAVE_QUDA
is set.ranmom
we download the momentum field, enabling residency using thequdaMomLoad
QUDA interface function.If the momentum is desired back on the host, then the function
qudaMomSave
can be used, which will invalidate the resident field, restoring host-device copying untilqudaMomLoad
is called again.This PR is dependent https://github.com/lattice/quda/pull/1014.