ALPAKA_ACCELERATOR_NAMESPACE::AlpakaService Class Reference

#include <AlpakaService.h>

Public Member Functions
	AlpakaService (edm::ParameterSet const &config, edm::ActivityRegistry &)
bool	enabled () const
	~AlpakaService ()

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

Private Attributes
bool	enabled_ = false
bool	verbose_ = false

Detailed Description

Definition at line 17 of file AlpakaService.h.

Constructor & Destructor Documentation

AlpakaService()

ALPAKA_ACCELERATOR_NAMESPACE::AlpakaService::AlpakaService	(	edm::ParameterSet const &	config,
		edm::ActivityRegistry &
	)

Definition at line 98 of file AlpakaService.cc.

References l1ct::clear(), cms::alpakatools::devices(), enabled_, ForwardName::getName(), dqmiodumpmetadata::n, Skims_PA_cff::name, or, MillePedeFileConverter_cfg::out, cms::alpakatools::platform(), AlCaHLTBitMon_QueryRunRegistry::string, ticlDumper_cff::suffix, and verbose_.

      : enabled_(config.getUntrackedParameter<bool>("enabled")),
        verbose_(config.getUntrackedParameter<bool>("verbose")) {
#ifdef ALPAKA_ACC_GPU_CUDA_ENABLED
    // rely on the CUDAService to initialise the CUDA devices
    edm::Service<CUDAInterface> cuda;
#endif  // ALPAKA_ACC_GPU_CUDA_ENABLED
#ifdef ALPAKA_ACC_GPU_HIP_ENABLED
    // rely on the ROCmService to initialise the ROCm devices
    edm::Service<ROCmInterface> rocm;
#endif  // ALPAKA_ACC_GPU_HIP_ENABLED

    // TODO from Andrea Bocci:
    //   - extract and print more information about the platform and devices

    if (not enabled_) {
      edm::LogInfo("AlpakaService") << ALPAKA_TYPE_ALIAS_NAME(AlpakaService) << " disabled by configuration";
      return;
    }

#ifdef ALPAKA_ACC_GPU_CUDA_ENABLED
    if (not cuda or not cuda->enabled()) {
      enabled_ = false;
      edm::LogInfo("AlpakaService") << ALPAKA_TYPE_ALIAS_NAME(AlpakaService) << " disabled by CUDAService";
      return;
    }
#endif  // ALPAKA_ACC_GPU_CUDA_ENABLED
#ifdef ALPAKA_ACC_GPU_HIP_ENABLED
    if (not rocm or not rocm->enabled()) {
      enabled_ = false;
      edm::LogInfo("AlpakaService") << ALPAKA_TYPE_ALIAS_NAME(AlpakaService) << " disabled by ROCmService";
      return;
    }
#endif  // ALPAKA_ACC_GPU_HIP_ENABLED

    // enumerate all devices on this platform
    auto const& devices = cms::alpakatools::devices<Platform>();
    if (devices.empty()) {
      const std::string platform = boost::core::demangle(typeid(Platform).name());
      edm::LogWarning("AlpakaService") << "Could not find any devices on platform " << platform << ".\n"
                                       << "Disabling " << ALPAKA_TYPE_ALIAS_NAME(AlpakaService) << ".";
      enabled_ = false;
      return;
    }

    {
      const char* suffix[] = {"s.", ":", "s:"};
      const auto n = devices.size();
      edm::LogInfo out("AlpakaService");
      out << ALPAKA_TYPE_ALIAS_NAME(AlpakaService) << " succesfully initialised.\n";
      out << "Found " << n << " device" << suffix[n < 2 ? n : 2];
      for (auto const& device : devices) {
        out << "\n  - " << alpaka::getName(device);
      }
    }

    // initialise the queue and event caches
    cms::alpakatools::getQueueCache<Queue>().clear();
    cms::alpakatools::getEventCache<Event>().clear();

    // initialise the caching memory allocators
    cms::alpakatools::AllocatorConfig hostAllocatorConfig =
        parseAllocatorConfig(config.getUntrackedParameter<edm::ParameterSet>("hostAllocator"));
    cms::alpakatools::getHostCachingAllocator<Queue>(hostAllocatorConfig, verbose_);
    cms::alpakatools::AllocatorConfig deviceAllocatorConfig =
        parseAllocatorConfig(config.getUntrackedParameter<edm::ParameterSet>("deviceAllocator"));
    for (auto const& device : devices)
      cms::alpakatools::getDeviceCachingAllocator<Device, Queue>(device, deviceAllocatorConfig, verbose_);
  }

~AlpakaService()

ALPAKA_ACCELERATOR_NAMESPACE::AlpakaService::~AlpakaService

(

)

Definition at line 168 of file AlpakaService.cc.

References l1ct::clear().

                                {
    // clean up the caching memory allocators
    cms::alpakatools::getHostCachingAllocator<Queue>().freeAllCached();
    for (auto const& device : cms::alpakatools::devices<Platform>())
      cms::alpakatools::getDeviceCachingAllocator<Device, Queue>(device).freeAllCached();

    // clean up the queue and event caches
    cms::alpakatools::getQueueCache<Queue>().clear();
    cms::alpakatools::getEventCache<Event>().clear();
  }

Member Function Documentation

enabled()

bool ALPAKA_ACCELERATOR_NAMESPACE::AlpakaService::enabled ( ) const

inline

Definition at line 24 of file AlpakaService.h.

References enabled_.

Referenced by ALPAKA_ACCELERATOR_NAMESPACE::detail::chooseDevice().

{ return enabled_; }

fillDescriptions()

void ALPAKA_ACCELERATOR_NAMESPACE::AlpakaService::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 179 of file AlpakaService.cc.

References edm::ConfigurationDescriptions::add(), and submitPVResolutionJobs::desc.

                                                                                 {
    edm::ParameterSetDescription desc;
    desc.addUntracked<bool>("enabled", true);
    desc.addUntracked<bool>("verbose", false);
    desc.addUntracked<edm::ParameterSetDescription>("hostAllocator", createAllocatorConfig())
        ->setComment("Configuration for the host's CachingAllocator");
    desc.addUntracked<edm::ParameterSetDescription>("deviceAllocator", createAllocatorConfig())
        ->setComment("Configuration for the devices' CachingAllocator");

    descriptions.add(ALPAKA_TYPE_ALIAS_NAME(AlpakaService), desc);
  }

Member Data Documentation

enabled_

bool ALPAKA_ACCELERATOR_NAMESPACE::AlpakaService::enabled_ = false

private

Definition at line 27 of file AlpakaService.h.

Referenced by AlpakaService(), and enabled().

verbose_

bool ALPAKA_ACCELERATOR_NAMESPACE::AlpakaService::verbose_ = false

private

Definition at line 28 of file AlpakaService.h.

Referenced by AlpakaService().