HeterogeneousCore/ROCmUtilities/interface/requireDevices.h - master - githubmirror/cmssw

HeterogeneousCore/ROCmUtilities/interface/requireDevices.h

19 строк569 B

Andrea Bocci

Implement requireDevices() for ROCm devices

30 янв 2023, 19:31

30 янв 2023, 19:317b3f444

#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