|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
#ifndef CUDADAtaFormats_HGCal_HGCUncalibRecHitHost_H
#define CUDADAtaFormats_HGCal_HGCUncalibRecHitHost_H
#include <cassert>
#include <numeric>
#include "HeterogeneousCore/CUDAUtilities/interface/host_unique_ptr.h"
#include "CUDADataFormats/HGCal/interface/HGCRecHitSoA.h"
#include "CUDADataFormats/HGCal/interface/HGCUncalibRecHitSoA.h"
template <class T>
class HGCUncalibRecHitHost {
public:
HGCUncalibRecHitHost() = default;
explicit HGCUncalibRecHitHost(uint32_t nhits, const T &hits, const cudaStream_t &stream) : nhits_(nhits) {
size_tot_ = std::accumulate(sizes_.begin(), sizes_.end(), 0); //this might be done at compile time
pad_ = ((nhits - 1) / 32 + 1) * 32; //align to warp boundary (assumption: warpSize = 32)
ptr_ = cms::cuda::make_host_unique<std::byte[]>(pad_ * size_tot_, stream);
defineSoAMemoryLayout_();
fillSoA_(hits);
}
~HGCUncalibRecHitHost() = default;
HGCUncalibRecHitHost(const HGCUncalibRecHitHost &) = delete;
HGCUncalibRecHitHost &operator=(const HGCUncalibRecHitHost &) = delete;
HGCUncalibRecHitHost(HGCUncalibRecHitHost &&) = default;
HGCUncalibRecHitHost &operator=(HGCUncalibRecHitHost &&) = default;
void defineSoAMemoryLayout_() {
soa_.amplitude_ = reinterpret_cast<float *>(ptr_.get());
soa_.pedestal_ = soa_.amplitude_ + pad_;
soa_.jitter_ = soa_.pedestal_ + pad_;
soa_.chi2_ = soa_.jitter_ + pad_;
soa_.OOTamplitude_ = soa_.chi2_ + pad_;
soa_.OOTchi2_ = soa_.OOTamplitude_ + pad_;
soa_.flags_ = reinterpret_cast<uint32_t *>(soa_.OOTchi2_ + pad_);
soa_.aux_ = soa_.flags_ + pad_;
soa_.id_ = soa_.aux_ + pad_;
soa_.aux_ = soa_.flags_ + pad_;
soa_.id_ = soa_.aux_ + pad_;
soa_.nbytes_ = size_tot_;
soa_.nhits_ = nhits_;
soa_.pad_ = pad_;
}
void fillSoA_(const T &c) {
for (unsigned i(0); i < nhits_; ++i) {
soa_.amplitude_[i] = c[i].amplitude();
soa_.pedestal_[i] = c[i].pedestal();
soa_.jitter_[i] = c[i].jitter();
soa_.chi2_[i] = c[i].chi2();
soa_.OOTamplitude_[i] = c[i].outOfTimeEnergy();
soa_.OOTchi2_[i] = c[i].outOfTimeChi2();
soa_.flags_[i] = c[i].flags();
soa_.aux_[i] = 0;
soa_.id_[i] = c[i].id().rawId();
}
}
HGCUncalibRecHitSoA get() const { return soa_; }
uint32_t nHits() const { return nhits_; }
uint32_t pad() const { return pad_; }
uint32_t nBytes() const { return size_tot_; }
private:
cms::cuda::host::unique_ptr<std::byte[]> ptr_;
HGCUncalibRecHitSoA soa_;
static constexpr std::array<int, memory::npointers::ntypes_hgcuncalibrechits_soa> sizes_ = {
{memory::npointers::float_hgcuncalibrechits_soa * sizeof(float),
memory::npointers::uint32_hgcuncalibrechits_soa * sizeof(uint32_t)}};
uint32_t size_tot_;
uint32_t pad_;
uint32_t nhits_;
};
#endif //CUDADAtaFormats_HGCal_HGCUncalibRecHitHost_H
|