Add separate constructor to StatePropagatorDataGpu for PME-only rank / PME tests

[alexxy/gromacs.git] / src / gromacs / ewald / tests / pmetestcommon.cpp

diff --git a/src/gromacs/ewald/tests/pmetestcommon.cpp b/src/gromacs/ewald/tests/pmetestcommon.cpp
index c82c0c685d93281de9b78ea65765bc210b70dfe0..2153e2c1395b554b397c784b099c49747b7bf0b8 100644 (file)
--- a/src/gromacs/ewald/tests/pmetestcommon.cpp
+++ b/src/gromacs/ewald/tests/pmetestcommon.cpp
@@ -172,7 +172,9 @@ std::unique_ptr<StatePropagatorDataGpu>
 makeStatePropagatorDataGpu(const gmx_pme_t &pme)
 {
     // TODO: Pin the host buffer and use async memory copies
-    return std::make_unique<StatePropagatorDataGpu>(pme_gpu_get_device_stream(&pme), nullptr, nullptr,
+    // TODO: Special constructor for PME-only rank / PME-tests is used here. There should be a mechanism to
+    //       restrict one from using other constructor here.
+    return std::make_unique<StatePropagatorDataGpu>(pme_gpu_get_device_stream(&pme),
                                                     pme_gpu_get_device_context(&pme),
                                                     GpuApiCallBehavior::Sync,
                                                     pme_gpu_get_padding_size(&pme));