HcalGainsGPU Class Reference

#include <HcalGainsGPU.h>

Classes
struct	Product

Public Member Functions
Product const &	getProduct (cudaStream_t) const
	HcalGainsGPU (HcalGains const &)
	~HcalGainsGPU ()=default

Private Attributes
cms::cuda::ESProduct< Product >	product_
uint64_t	totalChannels_
std::vector< float, cms::cuda::HostAllocator< float > >	values_

Detailed Description

Definition at line 13 of file HcalGainsGPU.h.

Constructor & Destructor Documentation

HcalGainsGPU()

HcalGainsGPU::HcalGainsGPU

(

HcalGains const &

gains

)

Definition at line 7 of file HcalGainsGPU.cc.

References ecalph2::gains.

    : totalChannels_{gains.getAllContainers()[0].second.size() + gains.getAllContainers()[1].second.size()},
      values_(totalChannels_ * 4) {
  auto const gainContainers = gains.getAllContainers();

  // fill in eb
  auto const& barrelValues = gainContainers[0].second;
  for (uint64_t i = 0; i < barrelValues.size(); ++i) {
    values_[i * 4] = barrelValues[i].getValue(0);
    values_[i * 4 + 1] = barrelValues[i].getValue(1);
    values_[i * 4 + 2] = barrelValues[i].getValue(2);
    values_[i * 4 + 3] = barrelValues[i].getValue(3);
  }

  // fill in ee
  auto const& endcapValues = gainContainers[1].second;
  auto const offset = barrelValues.size();
  for (uint64_t i = 0; i < endcapValues.size(); ++i) {
    auto const off = offset + i;
    values_[off * 4] = endcapValues[i].getValue(0);
    values_[off * 4 + 1] = endcapValues[i].getValue(1);
    values_[off * 4 + 2] = endcapValues[i].getValue(2);
    values_[off * 4 + 3] = endcapValues[i].getValue(3);
  }
}

~HcalGainsGPU()

HcalGainsGPU::~HcalGainsGPU ( )

default

Member Function Documentation

getProduct()

HcalGainsGPU::Product const & HcalGainsGPU::getProduct

(

cudaStream_t

stream

)

const

Definition at line 33 of file HcalGainsGPU.cc.

References cms::cuda::copyAsync(), product_, cms::cuda::stream, HcalGainsGPU::Product::values, and values_.

                                                                             {
  auto const& product =
      product_.dataForCurrentDeviceAsync(stream, [this](HcalGainsGPU::Product& product, cudaStream_t stream) {
        // allocate
        product.values = cms::cuda::make_device_unique<float[]>(values_.size(), stream);

        // transfer
        cms::cuda::copyAsync(product.values, values_, stream);
      });

  return product;
}

Member Data Documentation

product_

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

private

Definition at line 33 of file HcalGainsGPU.h.

Referenced by getProduct().

totalChannels_

uint64_t HcalGainsGPU::totalChannels_

private

Definition at line 30 of file HcalGainsGPU.h.

values_

std::vector<float, cms::cuda::HostAllocator<float> > HcalGainsGPU::values_

private

Definition at line 31 of file HcalGainsGPU.h.

Referenced by getProduct().