HeterogeneousCore/ROCmUtilities/src/requireDevices.cc - master - githubmirror/cmssw

HeterogeneousCore/ROCmUtilities/src/requireDevices.cc

30 строк677 B

Andrea Bocci

Implement requireDevices() for ROCm devices

30 янв 2023, 19:31

30 янв 2023, 19:317b3f444

#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