HeterogeneousCore/ROCmServices/interface/ROCmInterface.h - master - githubmirror/cmssw

HeterogeneousCore/ROCmServices/interface/ROCmInterface.h

19 строк545 B

Andrea Bocci

Add abstract interfaces for the CUDA and ROCm services

28 фев 2023, 15:35

28 фев 2023, 15:355ebe8d8

#ifndef HeterogeneousCore_ROCmServices_interface_ROCmInterface_h
#define HeterogeneousCore_ROCmServices_interface_ROCmInterface_h

#include <utility>

class ROCmInterface {
public:
  ROCmInterface() = default;
  virtual ~ROCmInterface() = default;

  virtual bool enabled() const = 0;

  virtual int numberOfDevices() const = 0;

  // Returns the (major, minor) compute capability of the given device.
  virtual std::pair<int, int> computeCapability(int device) const = 0;
};

#endif  // HeterogeneousCore_ROCmServices_interface_ROCmInterface_h