Implement requireDevices() for ROCm devices

HeterogeneousCore/ROCmUtilities/interface/requireDevices.h

@@ -0,0 +1,19 @@
+#ifndef HeterogeneousCore_ROCmUtilities_interface_requireDevices_h
+#define HeterogeneousCore_ROCmUtilities_interface_requireDevices_h
+
+/**
+ * These functions are meant to be called only from unit tests.
+ */
+namespace cms {
+  namespace rocmtest {
+
+    /// In presence of ROCm devices, return true; otherwise print message and return false
+    bool testDevices();
+
+    /// Print message and exit if there are no ROCm devices
+    void requireDevices();
+
+  }  // namespace rocmtest
+}  // namespace cms
+
+#endif  // HeterogeneousCore_ROCmUtilities_interface_requireDevices_h

HeterogeneousCore/ROCmUtilities/src/requireDevices.cc

@@ -0,0 +1,30 @@
+#include <cstdlib>
+#include <iostream>
+
+#include <hip/hip_runtime.h>
+
+#include "HeterogeneousCore/ROCmUtilities/interface/requireDevices.h"
+
+namespace cms::rocmtest {
+
+  bool testDevices() {
+    int devices = 0;
+    auto status = hipGetDeviceCount(&devices);
+    if (status != hipSuccess) {
+      std::cerr << "Failed to initialise the ROCm runtime, the test will be skipped.\n";
+      return false;
+    }
+    if (devices == 0) {
+      std::cerr << "No ROCm devices available, the test will be skipped.\n";
+      return false;
+    }
+    return true;
+  }
+
+  void requireDevices() {
+    if (not testDevices()) {
+      exit(EXIT_SUCCESS);
+    }
+  }
+
+}  // namespace cms::rocmtest

HeterogeneousCore/ROCmUtilities/test/BuildFile.xml

@@ -1,7 +1,13 @@
-<use name="HeterogeneousCore/ROCmUtilities"/>
 <iftool name="rocm">
   <bin file="testHipCheck.cpp" name="testHipCheck">
     <use name="catch2"/>
     <use name="rocm"/>
+    <use name="HeterogeneousCore/ROCmUtilities"/>
+  </bin>
+
+  <bin file="testRequireROCmDevices.cpp" name="testRequireROCmDevices">
+    <use name="catch2"/>
+    <use name="rocm"/>
+    <use name="HeterogeneousCore/ROCmUtilities"/>
   </bin>
 </iftool>

HeterogeneousCore/ROCmUtilities/test/testRequireROCmDevices.cpp

@@ -0,0 +1,21 @@
+// Catch2 headers
+#define CATCH_CONFIG_MAIN
+#include <catch.hpp>
+
+// ROCm headers
+#include <hip/hip_runtime.h>
+
+// CMSSW headers
+#include "HeterogeneousCore/ROCmUtilities/interface/hipCheck.h"
+#include "HeterogeneousCore/ROCmUtilities/interface/requireDevices.h"
+
+TEST_CASE("HeterogeneousCore/ROCmUtilities testRequireROCmDevices", "[testRequireROCmDevices]") {
+  SECTION("Test requireDevices()") {
+    cms::rocmtest::requireDevices();
+
+    int devices = 0;
+    hipCheck(hipGetDeviceCount(&devices));
+
+    REQUIRE(devices > 0);
+  }
+}