File indexing completed on 2023-03-17 11:18:59
0001 #ifndef RecoLocalCalo_HGCRecProducers_HeterogeneousHGCalHEFConditions_h
0002 #define RecoLocalCalo_HGCRecProducers_HeterogeneousHGCalHEFConditions_h
0003
0004 #include <numeric> //accumulate
0005 #include <typeinfo>
0006
0007 #include "FWCore/MessageLogger/interface/MessageLogger.h"
0008 #include "HeterogeneousCore/CUDACore/interface/ESProduct.h"
0009 #include "HeterogeneousCore/CUDAUtilities/interface/cudaCheck.h"
0010 #include "CUDADataFormats/HGCal/interface/HGCConditions.h"
0011 #include "Geometry/HGCalCommonData/interface/HGCalDDDConstants.h"
0012 #include "Geometry/HGCalCommonData/interface/HGCalParameters.h"
0013 #include "RecoLocalCalo/HGCalRecProducers/plugins/KernelManagerHGCalRecHit.h"
0014
0015 namespace cpar = hgcal_conditions::parameters;
0016
0017
0018
0019 class HeterogeneousHGCalHEFConditionsWrapper {
0020 public:
0021
0022
0023 HeterogeneousHGCalHEFConditionsWrapper(const HGCalParameters *);
0024
0025
0026 ~HeterogeneousHGCalHEFConditionsWrapper();
0027
0028
0029 hgcal_conditions::HeterogeneousHEFConditionsESProduct const *getHeterogeneousConditionsESProductAsync(
0030 cudaStream_t stream) const;
0031
0032 private:
0033
0034
0035 cpar::HeterogeneousHGCalHEFParameters params_;
0036
0037 std::vector<size_t> sizes_params_;
0038 size_t chunk_params_;
0039
0040 std::vector<size_t> calculate_memory_bytes_params_(const HGCalParameters *);
0041 size_t allocate_memory_params_(const std::vector<size_t> &);
0042 void transfer_data_to_heterogeneous_pointers_params_(const std::vector<size_t> &, const HGCalParameters *);
0043
0044
0045
0046 double *&select_pointer_d_(cpar::HeterogeneousHGCalHEFParameters *, const unsigned int &) const;
0047 std::vector<double> select_pointer_d_(const HGCalParameters *, const unsigned int &) const;
0048
0049 int32_t *&select_pointer_i_(cpar::HeterogeneousHGCalHEFParameters *, const unsigned int &) const;
0050 std::vector<int32_t> select_pointer_i_(const HGCalParameters *, const unsigned int &) const;
0051
0052
0053
0054 struct GPUData {
0055
0056 ~GPUData();
0057
0058 hgcal_conditions::HeterogeneousHEFConditionsESProduct *host = nullptr;
0059
0060 hgcal_conditions::HeterogeneousHEFConditionsESProduct *device = nullptr;
0061 };
0062
0063
0064
0065 cms::cuda::ESProduct<GPUData> gpuData_;
0066 };
0067
0068 #endif