Introduce DeviceStreamManager

[alexxy/gromacs.git] / src / gromacs / gpu_utils / tests / gputest.cpp

diff --git a/src/gromacs/gpu_utils/tests/gputest.cpp b/src/gromacs/gpu_utils/tests/gputest.cpp
index 2012d97da3b8bb56c3ba1f0e18154cb1eadf8bb0..a862361586ebf8b7b3d0e85e7630095a242d132c 100644 (file)
--- a/src/gromacs/gpu_utils/tests/gputest.cpp
+++ b/src/gromacs/gpu_utils/tests/gputest.cpp
@@ -1,7 +1,7 @@
 /*
  * This file is part of the GROMACS molecular simulation package.
  *
- * Copyright (c) 2017,2018,2019, by the GROMACS development team, led by
+ * Copyright (c) 2017,2018,2019,2020, by the GROMACS development team, led by
  * Mark Abraham, David van der Spoel, Berk Hess, and Erik Lindahl,
  * and including many others, as listed in the AUTHORS file in the
  * top-level source directory and at http://www.gromacs.org.
@@ -59,6 +59,7 @@ GpuTest::GpuTest()
     if (isGpuDetectionFunctional(nullptr))
     {
         findGpus(gpuInfo_);
+        compatibleGpuIds_ = getCompatibleGpus(*gpuInfo_);
     }
     // Failing to find valid GPUs does not require further action
 }
@@ -69,9 +70,20 @@ GpuTest::~GpuTest()
     sfree(gpuInfo_);
 }
 
-bool GpuTest::haveValidGpus() const
+bool GpuTest::haveCompatibleGpus() const
 {
-    return gpuInfo_->n_dev_compatible > 0;
+    return !compatibleGpuIds_.empty();
+}
+
+std::vector<const DeviceInformation*> GpuTest::getDeviceInfos() const
+{
+    std::vector<const DeviceInformation*> deviceInfos;
+    deviceInfos.reserve(compatibleGpuIds_.size());
+    for (const auto& id : compatibleGpuIds_)
+    {
+        deviceInfos.emplace_back(getDeviceInfo(*gpuInfo_, id));
+    }
+    return deviceInfos;
 }
 
 } // namespace test