CUDAService.h

Go to the documentation of this file.

#ifndef HeterogeneousCore_CUDAServices_CUDAService_h
#define HeterogeneousCore_CUDAServices_CUDAService_h
 
#include <utility>
#include <vector>
 
#include "FWCore/Utilities/interface/StreamID.h"
 
namespace edm {
  class ParameterSet;
  class ActivityRegistry;
  class ConfigurationDescriptions;
}  // namespace edm
 
class CUDAService {
public:
  CUDAService(edm::ParameterSet const& iConfig);
  ~CUDAService();
 
  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
 
  bool enabled() const { return enabled_; }
 
  int numberOfDevices() const { return numberOfDevices_; }
 
  // major, minor
  std::pair<int, int> computeCapability(int device) const { return computeCapabilities_.at(device); }
 
  // Returns the id of device with most free memory. If none is found, returns -1.
  int deviceWithMostFreeMemory() const;
 
private:
  int numberOfDevices_ = 0;
  std::vector<std::pair<int, int>> computeCapabilities_;
  bool enabled_ = false;
};
 
#endif