HeterogeneousCore/CUDAServices/interface/CUDAInterface.h - master - githubmirror/cmssw

HeterogeneousCore/CUDAServices/interface/CUDAInterface.h

19 строк544 B

Andrea Bocci

Add abstract interfaces for the CUDA and ROCm services

28 фев 2023, 15:35

28 фев 2023, 15:355ebe8d8

#ifndef HeterogeneousCore_CUDAServices_interface_CUDAInterface
#define HeterogeneousCore_CUDAServices_interface_CUDAInterface

#include <utility>

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

  virtual bool enabled() const = 0;

  virtual int numberOfDevices() const = 0;

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

#endif  // HeterogeneousCore_CUDAServices_interface_CUDAInterface