HcalRecoParamsGPU Class Reference

#include <HcalRecoParamsGPU.h>

Classes
struct	Product

Public Member Functions
Product const &	getProduct (cudaStream_t) const
	HcalRecoParamsGPU (HcalRecoParams const &)
	~HcalRecoParamsGPU ()=default

Private Attributes
std::vector< uint32_t, cms::cuda::HostAllocator < uint32_t > >	param1_
std::vector< uint32_t, cms::cuda::HostAllocator < uint32_t > >	param2_
cms::cuda::ESProduct< Product >	product_
uint64_t	totalChannels_

Detailed Description

Definition at line 14 of file HcalRecoParamsGPU.h.

Constructor & Destructor Documentation

HcalRecoParamsGPU::HcalRecoParamsGPU

(

HcalRecoParams const &

recoParams

)

Definition at line 7 of file HcalRecoParamsGPU.cc.

References HcalCondObjectContainer< Item >::getAllContainers().

    : totalChannels_{recoParams.getAllContainers()[0].second.size() + recoParams.getAllContainers()[1].second.size()},
      param1_(totalChannels_),
      param2_(totalChannels_) {
  auto const& containers = recoParams.getAllContainers();

  // fill in eb
  auto const& barrelValues = containers[0].second;
  for (uint64_t i = 0; i < barrelValues.size(); ++i) {
    param1_[i] = barrelValues[i].param1();
    param2_[i] = barrelValues[i].param2();
  }

  // fill in ee
  auto const& endcapValues = containers[1].second;
  auto const offset = barrelValues.size();
  for (uint64_t i = 0; i < endcapValues.size(); ++i) {
    param1_[i + offset] = endcapValues[i].param1();
    param2_[i + offset] = endcapValues[i].param2();
  }
}

HcalRecoParamsGPU::~HcalRecoParamsGPU ( )

default

Member Function Documentation

HcalRecoParamsGPU::Product const & HcalRecoParamsGPU::getProduct

(

cudaStream_t

stream

)

const

Definition at line 29 of file HcalRecoParamsGPU.cc.

References cms::cuda::copyAsync(), HcalRecoParamsGPU::Product::param1, param1_, HcalRecoParamsGPU::Product::param2, param2_, product_, and cms::cuda::stream.

                                                                                       {
  auto const& product =
      product_.dataForCurrentDeviceAsync(stream, [this](HcalRecoParamsGPU::Product& product, cudaStream_t stream) {
        // allocate
        product.param1 = cms::cuda::make_device_unique<uint32_t[]>(param1_.size(), stream);
        product.param2 = cms::cuda::make_device_unique<uint32_t[]>(param2_.size(), stream);

        // transfer
        cms::cuda::copyAsync(product.param1, param1_, stream);
        cms::cuda::copyAsync(product.param2, param2_, stream);
      });

  return product;
}

Member Data Documentation

std::vector<uint32_t, cms::cuda::HostAllocator<uint32_t> > HcalRecoParamsGPU::param1_

private

Definition at line 33 of file HcalRecoParamsGPU.h.

Referenced by getProduct().

std::vector<uint32_t, cms::cuda::HostAllocator<uint32_t> > HcalRecoParamsGPU::param2_

private

Definition at line 34 of file HcalRecoParamsGPU.h.

Referenced by getProduct().

cms::cuda::ESProduct<Product> HcalRecoParamsGPU::product_

private

Definition at line 36 of file HcalRecoParamsGPU.h.

Referenced by getProduct().

uint64_t HcalRecoParamsGPU::totalChannels_

private

Definition at line 32 of file HcalRecoParamsGPU.h.